onnx_diagnostic.reference.torch_ops.shape_ops

class onnx_diagnostic.reference.torch_ops.shape_ops.ConstantOfShape_9(node: NodeProto, version: int | None = None, device: device | None = None)[source][source]

ConstantOfShape

classmethod device_dependent() bool[source][source]

Returns True if the kernel needs a device to be efficiently initialized.

run(shape: OpRunTensor) OpRunTensor[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Expand_8(node: NodeProto, version: int | None = None)[source][source]

Expand

run(data: OpRunTensor, shape: OpRunTensor) OpRunTensor[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Reshape_14(node: NodeProto, version: int | None = None)[source][source]

Reshape

run(data: OpRunTensor, shape: OpRunTensor) OpRunTensor[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Shape_15(node: NodeProto, version: int | None = None)[source][source]
run(data: OpRunTensor) OpRunTensor[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Split_18(node: NodeProto, version: int | None = None)[source][source]
run(data: OpRunTensor, split: OpRunTensor | None = None) Tuple[OpRunTensor, ...][source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Squeeze_13(node: NodeProto, version: int | None = None)[source][source]

Squeeze

run(data: OpRunTensor, axes: OpRunTensor | None = None) OpRunTensor[source][source]

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.shape_ops.Unsqueeze_13(node: NodeProto, version: int | None = None)[source][source]

Unsqueeze

run(data: OpRunTensor, axes: OpRunTensor) OpRunTensor[source][source]

Kernel implementation.