onnx_diagnostic.reference.torch_ops.nn_ops

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

AveragePool

run(x)[source][source]

Kernel implementation.

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

Conv

run(x, w, b=None)[source][source]

Kernel implementation.

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

LayerNormalization

run(x, scale, bias=None)[source][source]

Kernel implementation.

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

Softmax

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

Kernel implementation.

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

Tanh

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

Kernel implementation.