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.