module onnxrt.ops_shape#

Short summary#

module mlprodict.onnxrt.ops_shape

Shortcut to ops_shape.

source on GitHub

Functions#

function

truncated documentation

shape_dispatch

Calls the corresponding fucntion for every node.

Documentation#

Shortcut to ops_shape.

source on GitHub

mlprodict.onnxrt.ops_shape.shape_dispatch(cache, known_shape, node, rt_class=None)#

Calls the corresponding fucntion for every node.

Parameters:

  • cache – cache used function

  • known_shape – known_shape for all results

  • node – onnx node

  • rt_class – a node may be a predefined function in onnx, if no specific function is available, the predefined onnx definition is used and run through this runtime

Returns:

was known_shape updated or not…

source on GitHub