onnx_diagnostic.reference.torch_ops.nn_ops

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

AveragePool

run(x)

Kernel implementation.

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

Conv

run(x, w, b=None)

Kernel implementation.

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

LayerNormalization

run(x, scale, bias=None)

Kernel implementation.

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

Softmax

run(data: OpRunTensor) → OpRunTensor

Kernel implementation.

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

Tanh

run(data: OpRunTensor) → OpRunTensor

Kernel implementation.