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.

source on GitHub

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)#

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

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.