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.