module onnxrt.ops_cpu#

Short summary#

module mlprodict.onnxrt.ops_cpu

Shortcut to ops_cpu.

source on GitHub

Functions#

function

truncated documentation

load_op

Gets the operator related to the onnx node.

register_operator

Registers a new runtime operator.

Documentation#

Shortcut to ops_cpu.

source on GitHub

mlprodict.onnxrt.ops_cpu.load_op(onnx_node, desc=None, options=None, runtime=None)#

Gets the operator related to the onnx node.

Parameters:

  • onnx_node – onnx node

  • desc – internal representation

  • options – runtime options

  • runtime – runtime

  • existing_functions – existing functions

Returns:

runtime class

source on GitHub

mlprodict.onnxrt.ops_cpu.register_operator(cls, name=None, overwrite=True)#

Registers a new runtime operator.

Parameters:

  • cls – class

  • name – by default cls.__name__, or name if defined

  • overwrite – overwrite or raise an exception

source on GitHub