module onnxrt.ops_shape._element_wise#

Short summary#

module mlprodict.onnxrt.ops_shape._element_wise

Computes shape inference for element wise operators.

source on GitHub

Functions#

function

truncated documentation

_element_wise

Infers shape for an element wise operator. The function returns but updates known_shapes.

shape_add

Infers shape for operator Add.

shape_and

Infers shape for operator And.

shape_div

Infers shape for operator Div.

shape_equal

Infers shape for operator Equal.

shape_greater

Infers shape for operator Greater.

shape_greaterorequal

Infers shape for operator GreaterOrEqual.

shape_less

Infers shape for operator Less.

shape_lessorequal

Infers shape for operator LessOrEqual.

shape_max

Infers shape for operator Max.

shape_min

Infers shape for operator Min.

shape_mod

Infers shape for operator Mod.

shape_mul

Infers shape for operator Mul.

shape_or

Infers shape for operator Or.

shape_pow

Infers shape for operator Pow.

shape_sub

Infers shape for operator Sub.

shape_xor

Infers shape for operator Xor.

Documentation#

Computes shape inference for element wise operators.

source on GitHub

mlprodict.onnxrt.ops_shape._element_wise._element_wise(known_shapes, node, return_bool=False, same_type=True, one_input=False)#

Infers shape for an element wise operator. The function returns but updates known_shapes.

Parameters:

  • known_shapes – known shapes

  • node – Onnx node

  • return_bool – return boolean

  • same_type – check the type are the same

  • one_input – allow one input

Returns:

updated or not

source on GitHub

mlprodict.onnxrt.ops_shape._element_wise.shape_add(known_shapes, node)#

Infers shape for operator Add.

mlprodict.onnxrt.ops_shape._element_wise.shape_and(known_shapes, node)#

Infers shape for operator And.

mlprodict.onnxrt.ops_shape._element_wise.shape_div(known_shapes, node)#

Infers shape for operator Div.

mlprodict.onnxrt.ops_shape._element_wise.shape_equal(known_shapes, node)#

Infers shape for operator Equal.

mlprodict.onnxrt.ops_shape._element_wise.shape_greater(known_shapes, node)#

Infers shape for operator Greater.

mlprodict.onnxrt.ops_shape._element_wise.shape_greaterorequal(known_shapes, node)#

Infers shape for operator GreaterOrEqual.

mlprodict.onnxrt.ops_shape._element_wise.shape_less(known_shapes, node)#

Infers shape for operator Less.

mlprodict.onnxrt.ops_shape._element_wise.shape_lessorequal(known_shapes, node)#

Infers shape for operator LessOrEqual.

mlprodict.onnxrt.ops_shape._element_wise.shape_max(known_shapes, node)#

Infers shape for operator Max.

mlprodict.onnxrt.ops_shape._element_wise.shape_min(known_shapes, node)#

Infers shape for operator Min.

mlprodict.onnxrt.ops_shape._element_wise.shape_mod(known_shapes, node)#

Infers shape for operator Mod.

mlprodict.onnxrt.ops_shape._element_wise.shape_mul(known_shapes, node)#

Infers shape for operator Mul.

mlprodict.onnxrt.ops_shape._element_wise.shape_or(known_shapes, node)#

Infers shape for operator Or.

mlprodict.onnxrt.ops_shape._element_wise.shape_pow(known_shapes, node)#

Infers shape for operator Pow.

mlprodict.onnxrt.ops_shape._element_wise.shape_sub(known_shapes, node)#

Infers shape for operator Sub.

mlprodict.onnxrt.ops_shape._element_wise.shape_xor(known_shapes, node)#

Infers shape for operator Xor.