module onnx_conv.register_rewritten_converters#

Short summary#

module mlprodict.onnx_conv.register_rewritten_converters

Rewrites some of the converters implemented in sklearn-onnx.

source on GitHub

Functions#

function

truncated documentation

register_new_operators

Registers new operator relying on pieces implemented in this package such as the numpy API for ONNX.

register_rewritten_operators

Registers modified operators and returns the old values.

Documentation#

Rewrites some of the converters implemented in sklearn-onnx.

source on GitHub

mlprodict.onnx_conv.register_rewritten_converters.register_new_operators()#

Registers new operator relying on pieces implemented in this package such as the numpy API for ONNX.

source on GitHub

mlprodict.onnx_conv.register_rewritten_converters.register_rewritten_operators(new_converters=None, new_shape_calculators=None)#

Registers modified operators and returns the old values.

Parameters:
  • new_converters – converters to rewrite or None to rewrite default ones

  • new_shape_calculators – shape calculators to rewrite or None to rewrite default ones

Returns:

old converters, old shape calculators

source on GitHub