yobx.litert.ops.elementwise#

Converters for TFLite element-wise binary and unary ops.

yobx.litert.ops.elementwise.convert_abs(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite ABS → ONNX Abs.

yobx.litert.ops.elementwise.convert_add(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite ADD → ONNX Add.

yobx.litert.ops.elementwise.convert_ceil(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite CEIL → ONNX Ceil.

yobx.litert.ops.elementwise.convert_div(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite DIV → ONNX Div.

yobx.litert.ops.elementwise.convert_exp(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite EXP → ONNX Exp.

yobx.litert.ops.elementwise.convert_floor(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite FLOOR → ONNX Floor.

yobx.litert.ops.elementwise.convert_floor_div(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite FLOOR_DIV → ONNX Floor(Div(a, b)).

yobx.litert.ops.elementwise.convert_log(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite LOG → ONNX Log.

yobx.litert.ops.elementwise.convert_logical_and(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite LOGICAL_AND → ONNX And.

yobx.litert.ops.elementwise.convert_logical_not(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite LOGICAL_NOT → ONNX Not.

yobx.litert.ops.elementwise.convert_logical_or(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite LOGICAL_OR → ONNX Or.

yobx.litert.ops.elementwise.convert_mul(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite MUL → ONNX Mul.

yobx.litert.ops.elementwise.convert_neg(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite NEG → ONNX Neg.

yobx.litert.ops.elementwise.convert_pow(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite POW → ONNX Pow.

yobx.litert.ops.elementwise.convert_round(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite ROUND → ONNX Round.

yobx.litert.ops.elementwise.convert_rsqrt(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite RSQRT → ONNX Reciprocal(Sqrt(x)).

yobx.litert.ops.elementwise.convert_sin(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite SIN → ONNX Sin.

yobx.litert.ops.elementwise.convert_sqrt(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite SQRT → ONNX Sqrt.

yobx.litert.ops.elementwise.convert_squared_difference(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite SQUARED_DIFFERENCE → ONNX Pow(Sub(a, b), 2).

yobx.litert.ops.elementwise.convert_sub(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: TFLiteOperator) → str[source]#: TFLite SUB → ONNX Sub.