documentation (#137)

* documentation

* fix graph

Author: xadupre

_doc/technical/plot_layer_norm_discrepancies.py

@@ -143,7 +143,7 @@ def cast_feeds(itype, provider, feeds):
 # %%
 # Visually.
 
-df["abs"].plot(title="Discrepancies ORT / torch for LayerNorm(X) @ W + B")
+df["abs"].plot.bar(title="Discrepancies ORT / torch for LayerNorm(X) @ W + B")
 
 # %%
 # The discrepancies are significant on CUDA, higher for float16.
@@ -207,4 +207,6 @@ def cast_feeds(itype, provider, feeds):
 # %%
 # Visually.
 
-df[["diff_ort", "diff_torch"]].plot(title="ORT/Torch or Torch/ORT for LayerNorm(X) @ W + B")
+df[["diff_ort", "diff_torch"]].plot.bar(
+    title="ORT/Torch or Torch/ORT for LayerNorm(X) @ W + B"
+)

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(