onnx_diagnostic.reference.torch_ops.access_ops

class onnx_diagnostic.reference.torch_ops.access_ops.Gather_1(node: NodeProto, version: int | None = None, verbose: int = 0)

Gather

run(x, indices)

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.access_ops.ScatterND_16(node: NodeProto, version: int | None = None, verbose: int = 0)

ScatterND

run(data: OpRunTensor, indices: OpRunTensor, updates: OpRunTensor) → OpRunTensor

Kernel implementation.

class onnx_diagnostic.reference.torch_ops.access_ops.Slice_13(node: NodeProto, version: int | None = None, verbose: int = 0, custom_kernels: Dict[Tuple[str, str], type] | None = None)

Slice

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

Kernel implementation.