module onnxrt.ops_onnxruntime._op#

Inheritance diagram of mlprodict.onnxrt.ops_onnxruntime._op

Short summary#

module mlprodict.onnxrt.ops_onnxruntime._op

Shortcut to ops_onnxruntime.

source on GitHub

Classes#

class

truncated documentation

OpRunOnnxRuntime

Unique operator which calls onnxruntime to compute predictions for one operator.

Methods#

method

truncated documentation

__init__

_guess_proto_type

_init

Initializes the node.

_name_mapping

need_context

Tells the runtime if this node needs the context (all the results produced so far) as it may silently access …

run

Should be overwritten.

Documentation#

Shortcut to ops_onnxruntime.

source on GitHub

class mlprodict.onnxrt.ops_onnxruntime._op.OpRunOnnxRuntime(onnx_node, desc=None, variables=None, dtype=None, runtime=None, **options)#

Bases: object

Unique operator which calls onnxruntime to compute predictions for one operator.

source on GitHub

Parameters:

  • onnx_node – onnx node

  • desc – internal representation

  • variables – registered variables created by previous operators

  • dtype – float computation type

  • options – runtime options

  • runtime – onnxruntime1, onnxruntime1-cuda, …

source on GitHub

__init__(onnx_node, desc=None, variables=None, dtype=None, runtime=None, **options)#
Parameters:

  • onnx_node – onnx node

  • desc – internal representation

  • variables – registered variables created by previous operators

  • dtype – float computation type

  • options – runtime options

  • runtime – onnxruntime1, onnxruntime1-cuda, …

source on GitHub

_guess_proto_type(dtype)#
_init(variables=None)#

Initializes the node.

Parameters:

variables – registered variables created by previous operators

The current implementation for operator Scan only works for matrices.

source on GitHub

_name_mapping(inputs)#
need_context()#

Tells the runtime if this node needs the context (all the results produced so far) as it may silently access one of them (operator Loop). The default answer is False.

source on GitHub

run(*args, **kwargs)#

Should be overwritten.

source on GitHub