onnx_diagnostic.helpers.doc_helper¶

class onnx_diagnostic.helpers.doc_helper.LayerNormalizationOrt(node: NodeProto, version=None, device: device | None = None, verbose: int = 0)[source][source]¶

LayerNormalization with onnxruntime

classmethod device_dependent() → bool[source][source]¶: Needs device.

run(x, scale, bias=None)[source][source]¶: Kernel implementation.

class onnx_diagnostic.helpers.doc_helper.MatMulOrt(node: NodeProto, version=None, device: device | None = None, verbose: int = 0)[source][source]¶

MatMul with onnxruntime

classmethod device_dependent() → bool[source][source]¶: Needs device.

run(a, b)[source][source]¶: Kernel implementation.