fix documentation (#121)

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.6.1
 +++++
 
+* :pr:`120`: enables TorchOnnxEvaluator in command line ``python -m onnx_diagnostic validate ...``
 * :pr:`115`, :pr:`116`, :pr:`117`, :pr:`118`, :pr:`119`:
   first steps for TorchOnnxEvaluator
 * :pr:`114`: extends the list of known rewritings

onnx_diagnostic/reference/torch_evaluator.py

@@ -58,7 +58,7 @@ class TorchOnnxEvaluator:
 
     The class is not multithreaded. `runtime_info` gets updated
     by the the class. The list of available kernels is returned by function
-    :func:`get_kernels`.
+    :func:`onnx_diagnostic.reference.torch_evaluator.get_kernels`.
     """
 
     def __init__(

onnx_diagnostic/torch_export_patches/patch_module_helper.py

@@ -67,11 +67,11 @@ def known_transformers_rewritings_clamp_float16() -> Dict[str, str]:
         :showcode:
 
         import pprint
-        from onnx_diagnostic.torch_export_patches.patch_model_helper import (
-            known_transformers_rewritings,
+        from onnx_diagnostic.torch_export_patches.patch_module_helper import (
+            known_transformers_rewritings_clamp_float16,
         )
 
-        pprint.pprint(known_transformers_rewritings())
+        pprint.pprint(known_transformers_rewritings_clamp_float16())
     """
     _alias = {
         "AutoformerEncoder": "AutoformerEncoderLayer",
@@ -130,13 +130,13 @@ def rewritings_transformers_clamp_float16(cls_name) -> List[type]:
         :showcode:
 
         import pprint
-        from onnx_diagnostic.torch_export_patches.patch_model_helper import (
+        from onnx_diagnostic.torch_export_patches.patch_module_helper import (
             _rewrite_forward_clamp_float16,
         )
 
-        pprint.pprint(_rewrite_forward_clamp_float16()
+        pprint.pprint(_rewrite_forward_clamp_float16())
 
-    Function :func:`known_transformers_rewritings` collects
+    Function `_rewrite_forward_clamp_float16` collects
     all model classes using those layers.
     """
     _known = _rewrite_forward_clamp_float16()
@@ -161,7 +161,7 @@ def _add(f):
 def code_needing_rewriting(cls_name: str) -> Optional[List[Any]]:
     """
     Returns a known list of classes mapped to a known rewritings
-    because of control flow. See :func:`registered_transformers_rewritings`.
+    because of control flow. See :func:`known_transformers_rewritings_clamp_float16`.
 
     :param cls_name: name of the class
     :return: a list of rewriting