module `onnxrt.ops_shape._element_unary`#

Short summary#

module mlprodict.onnxrt.ops_shape._element_unary

Computes shape inference for element wise operators with one input.

Functions#

function	truncated documentation
`_element_unary`	Infers shape for an element wise operator. The function returns but updates known_shapes.
`shape_abs`	Infers shape for operator Abs.
`shape_acos`	Infers shape for operator Acos.
`shape_acosh`	Infers shape for operator Acosh.
`shape_asin`	Infers shape for operator Asin.
`shape_asinh`	Infers shape for operator Asinh.
`shape_atan`	Infers shape for operator Atan.
`shape_atanh`	Infers shape for operator Atanh.
`shape_castlike`	Infers shape for operator CastLike.
`shape_ceil`	Infers shape for operator Ceil.
`shape_celu`	Infers shape for operator Celu.
`shape_clip`	Infers shape for operator Clip.
`shape_cos`	Infers shape for operator Cos.
`shape_cosh`	Infers shape for operator Cosh.
`shape_elu`	Infers shape for operator Elu.
`shape_erf`	Infers shape for operator Erf.
`shape_exp`	Infers shape for operator Exp.
`shape_floor`	Infers shape for operator Floor.
`shape_hardmax`	Infers shape for operator Hardmax.
`shape_hardsigmoid`	Infers shape for operator HardSigmoid.
`shape_identity`	Infers shape for operator Identity.
`shape_isinf`	Infers shape for operator IsInf.
`shape_isnan`	Infers shape for operator IsNan.
`shape_leakyrelu`	Infers shape for operator LeakyRelu.
`shape_log`	Infers shape for operator Log.
`shape_logsoftmax`	Infers shape for operator LogSoftmax.
`shape_neg`	Infers shape for operator Neg.
`shape_not`	Infers shape for operator Not.
`shape_reciprocal`	Infers shape for operator Reciprocal.
`shape_relu`	Infers shape for operator Relu.
`shape_round`	Infers shape for operator Round.
`shape_selu`	Infers shape for operator Selu.
`shape_shrink`	Infers shape for operator Shrink.
`shape_sigmoid`	Infers shape for operator Sigmoid.
`shape_sign`	Infers shape for operator Sigmoid.
`shape_sin`	Infers shape for operator Sin.
`shape_sinh`	Infers shape for operator Sinh.
`shape_softmax`	Infers shape for operator Softmax.
`shape_softplus`	Infers shape for operator Softplus.
`shape_softsign`	Infers shape for operator Softsign.
`shape_sqrt`	Infers shape for operator Sqrt.
`shape_tan`	Infers shape for operator Tan.
`shape_tanh`	Infers shape for operator Tanh.
`shape_thresholdedrelu`	Infers shape for operator ThresholdedRelu.
`shape_trilu`	Infers shape for operator Trilu.

Documentation#

Computes shape inference for element wise operators with one input.

source on GitHub

mlprodict.onnxrt.ops_shape._element_unary._element_unary(known_shapes, node, dtype=None, one_input=True)#

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

Parameters:

known_shapes – known shapes
node – Onnx node
dtype – None to keep the same type as input, not None to change it
one_input – check there is only one input

Returns:

updated or not

source on GitHub

mlprodict.onnxrt.ops_shape._element_unary.shape_abs(known_shapes, node)#: Infers shape for operator Abs.

mlprodict.onnxrt.ops_shape._element_unary.shape_acos(known_shapes, node)#: Infers shape for operator Acos.

mlprodict.onnxrt.ops_shape._element_unary.shape_acosh(known_shapes, node)#: Infers shape for operator Acosh.

mlprodict.onnxrt.ops_shape._element_unary.shape_asin(known_shapes, node)#: Infers shape for operator Asin.

mlprodict.onnxrt.ops_shape._element_unary.shape_asinh(known_shapes, node)#: Infers shape for operator Asinh.

mlprodict.onnxrt.ops_shape._element_unary.shape_atan(known_shapes, node)#: Infers shape for operator Atan.

mlprodict.onnxrt.ops_shape._element_unary.shape_atanh(known_shapes, node)#: Infers shape for operator Atanh.

mlprodict.onnxrt.ops_shape._element_unary.shape_castlike(known_shapes, node)#: Infers shape for operator CastLike.

mlprodict.onnxrt.ops_shape._element_unary.shape_ceil(known_shapes, node)#: Infers shape for operator Ceil.

mlprodict.onnxrt.ops_shape._element_unary.shape_celu(known_shapes, node)#: Infers shape for operator Celu.

mlprodict.onnxrt.ops_shape._element_unary.shape_clip(known_shapes, node)#: Infers shape for operator Clip.

mlprodict.onnxrt.ops_shape._element_unary.shape_cos(known_shapes, node)#: Infers shape for operator Cos.

mlprodict.onnxrt.ops_shape._element_unary.shape_cosh(known_shapes, node)#: Infers shape for operator Cosh.

mlprodict.onnxrt.ops_shape._element_unary.shape_elu(known_shapes, node)#: Infers shape for operator Elu.

mlprodict.onnxrt.ops_shape._element_unary.shape_erf(known_shapes, node)#: Infers shape for operator Erf.

mlprodict.onnxrt.ops_shape._element_unary.shape_exp(known_shapes, node)#: Infers shape for operator Exp.

mlprodict.onnxrt.ops_shape._element_unary.shape_floor(known_shapes, node)#: Infers shape for operator Floor.

mlprodict.onnxrt.ops_shape._element_unary.shape_hardmax(known_shapes, node)#: Infers shape for operator Hardmax.

mlprodict.onnxrt.ops_shape._element_unary.shape_hardsigmoid(known_shapes, node)#: Infers shape for operator HardSigmoid.

mlprodict.onnxrt.ops_shape._element_unary.shape_identity(known_shapes, node)#: Infers shape for operator Identity.

mlprodict.onnxrt.ops_shape._element_unary.shape_isinf(known_shapes, node)#: Infers shape for operator IsInf.

mlprodict.onnxrt.ops_shape._element_unary.shape_isnan(known_shapes, node)#: Infers shape for operator IsNan.

mlprodict.onnxrt.ops_shape._element_unary.shape_leakyrelu(known_shapes, node)#: Infers shape for operator LeakyRelu.

mlprodict.onnxrt.ops_shape._element_unary.shape_log(known_shapes, node)#: Infers shape for operator Log.

mlprodict.onnxrt.ops_shape._element_unary.shape_logsoftmax(known_shapes, node)#: Infers shape for operator LogSoftmax.

mlprodict.onnxrt.ops_shape._element_unary.shape_neg(known_shapes, node)#: Infers shape for operator Neg.

mlprodict.onnxrt.ops_shape._element_unary.shape_not(known_shapes, node)#: Infers shape for operator Not.

mlprodict.onnxrt.ops_shape._element_unary.shape_reciprocal(known_shapes, node)#: Infers shape for operator Reciprocal.

mlprodict.onnxrt.ops_shape._element_unary.shape_relu(known_shapes, node)#: Infers shape for operator Relu.

mlprodict.onnxrt.ops_shape._element_unary.shape_round(known_shapes, node)#: Infers shape for operator Round.

mlprodict.onnxrt.ops_shape._element_unary.shape_selu(known_shapes, node)#: Infers shape for operator Selu.

mlprodict.onnxrt.ops_shape._element_unary.shape_shrink(known_shapes, node)#: Infers shape for operator Shrink.

mlprodict.onnxrt.ops_shape._element_unary.shape_sigmoid(known_shapes, node)#: Infers shape for operator Sigmoid.

mlprodict.onnxrt.ops_shape._element_unary.shape_sign(known_shapes, node)#: Infers shape for operator Sigmoid.

mlprodict.onnxrt.ops_shape._element_unary.shape_sin(known_shapes, node)#: Infers shape for operator Sin.

mlprodict.onnxrt.ops_shape._element_unary.shape_sinh(known_shapes, node)#: Infers shape for operator Sinh.

mlprodict.onnxrt.ops_shape._element_unary.shape_softmax(known_shapes, node)#: Infers shape for operator Softmax.

mlprodict.onnxrt.ops_shape._element_unary.shape_softplus(known_shapes, node)#: Infers shape for operator Softplus.

mlprodict.onnxrt.ops_shape._element_unary.shape_softsign(known_shapes, node)#: Infers shape for operator Softsign.

mlprodict.onnxrt.ops_shape._element_unary.shape_sqrt(known_shapes, node)#: Infers shape for operator Sqrt.

mlprodict.onnxrt.ops_shape._element_unary.shape_tan(known_shapes, node)#: Infers shape for operator Tan.

mlprodict.onnxrt.ops_shape._element_unary.shape_tanh(known_shapes, node)#: Infers shape for operator Tanh.

mlprodict.onnxrt.ops_shape._element_unary.shape_thresholdedrelu(known_shapes, node)#: Infers shape for operator ThresholdedRelu.

mlprodict.onnxrt.ops_shape._element_unary.shape_trilu(known_shapes, node)#: Infers shape for operator Trilu.

module onnxrt.ops_shape._element_unary#

Short summary#

Functions#

Documentation#

module `onnxrt.ops_shape._element_unary`#