onnx_diagnostic.reference.torch_ops.access_ops

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

Gather

run(x, indices)[source][source]

Kernel implementation.

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

ScatterND

run(data: OpRunTensor, indices: OpRunTensor, updates: OpRunTensor) → OpRunTensor[source][source]

Kernel implementation.

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

Slice

run(data: OpRunTensor, starts: OpRunTensor, ends: OpRunTensor, axes: OpRunTensor | None = None, steps: OpRunTensor | None = None) → OpRunTensor[source][source]

Kernel implementation.