onnx_diagnostic.reference.torch_ops.other_ops¶
- class onnx_diagnostic.reference.torch_ops.other_ops.CastLike_15(node: NodeProto, version: int | None = None)[source][source]¶
Cast
- run(data: OpRunTensor, like: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Cast_6(node: NodeProto, version: int | None = None)[source][source]¶
Cast
- run(data: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Concat_1(node: NodeProto, version: int | None = None)[source][source]¶
Concat
- run(*data: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.NonZero_13(node: NodeProto, version: int | None = None)[source][source]¶
NonZero
- run(x: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Tile_6(node: NodeProto, version: int | None = None)[source][source]¶
Tile
- run(x: OpRunTensor, repeat: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Transpose_1(node: NodeProto, version: int | None = None)[source][source]¶
Transpose
- run(data: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Trilu_14(node: NodeProto, version: int | None = None)[source][source]¶
Trilu
- run(data: OpRunTensor, k: OpRunTensor | None = None) OpRunTensor [source][source]¶
Kernel implementation.
- class onnx_diagnostic.reference.torch_ops.other_ops.Where_9(node: NodeProto, version: int | None = None)[source][source]¶
Where
- run(cond: OpRunTensor, x: OpRunTensor, y: OpRunTensor) OpRunTensor [source][source]¶
Kernel implementation.