documentation

Author: xadupre

onnx_diagnostic/helpers/doc_helper.py

@@ -20,7 +20,7 @@ def __init__(
         node: onnx.NodeProto,
         version=None,
         device: Optional[torch.device] = None,
-        verbose=0,
+        verbose: int = 0,
     ):
         super().__init__(node, version, verbose=verbose)
         self.axis = self.get_attribute_int(node, "axis", -1)
@@ -101,7 +101,7 @@ def __init__(
         node: onnx.NodeProto,
         version=None,
         device: Optional[torch.device] = None,
-        verbose=0,
+        verbose: int = 0,
     ):
         super().__init__(node, version, verbose=verbose)
         self.device = device

onnx_diagnostic/reference/torch_evaluator.py

@@ -168,8 +168,8 @@ class TorchOnnxEvaluator:
         class LayerNormalizationOrt(OpRunKernel):
             "LayerNormalization based on onnxruntime"
 
-            def __init__(self, node: onnx.NodeProto, version=None):
-                super().__init__(node, version)
+            def __init__(self, node: onnx.NodeProto, version=None, verbose=0):
+                super().__init__(node, version, verbose=verbose)
                 self.axis = self.get_attribute_int(node, "axis", -1)
                 self.epsilon = self.get_attribute_float(node, "epsilon", 1e-5)
                 self.stash_type = onnx_dtype_to_torch_dtype(