documentation

Author: xadupre

_doc/status/exported_program_dynamic.rst

@@ -15,7 +15,8 @@ to the original model.
     import inspect
     import textwrap
     import pandas
-    from onnx_diagnostic.torch_export_patch.eval import discover, run_exporter
+    from onnx_diagnostic.helpers import string_type
+    from onnx_diagnostic.torch_export_patches.eval import discover, run_exporter
     from onnx_diagnostic.ext_test_case import unit_test_going
 
     cases = discover()

_doc/status/index.rst

@@ -2,10 +2,11 @@
 Exporter Status
 ===============
 
-Following sections tries to capture what works and what does not.
+Following sections tries to capture what patches are put in place to export,
+what works and what does not with :func:`torch.export.export`.
 
 .. toctree::
     :maxdepth: 1
 
     exported_program_dynamic
-    patched_classes
+    patches_coverage

_doc/status/patches_for_transformers.rst renamed to _doc/status/patches_coverage.rst

@@ -1,6 +1,6 @@
-========================
-Patches for transformers
-========================
+=======================
+Coverage of the Patches
+=======================
 
 Serialized Classes
 ==================
@@ -10,10 +10,9 @@ The following code shows the list of serialized classes in transformers.
 .. runpython::
     :showcode:
 
-    import pprint
-    import onnx_diagnostic.torch_export_patch.patches.patch_transformers as p
+    import onnx_diagnostic.torch_export_patches.onnx_export_serialization as p
 
-    pprint.pprint(sorted(p.serialization_functions()))
+    print('\n'.join(sorted(p.serialization_functions())))
 
 Patched Classes
 ===============
@@ -24,7 +23,7 @@ for transformers.
 .. runpython::
     :showcode:
 
-    import onnx_diagnostic.torch_export_patch.patches.patch_transformers as p
+    import onnx_diagnostic.torch_export_patches.patches.patch_transformers as p
 
     for name, cls in p.__dict__.items():
         if name.startswith("patched_"):
@@ -39,7 +38,7 @@ for transformers.
 .. runpython::
     :showcode:
 
-    import onnx_diagnostic.torch_export_patch.patches.patch_module_helper as p
+    import onnx_diagnostic.torch_export_patches.patch_module_helper as p
 
     for name, f in p.__dict__.items():
         if name.startswith("_rewrite_"):

onnx_diagnostic/torch_export_patches/eval/__init__.py

@@ -15,7 +15,7 @@ def discover():
         :showcode:
 
         import pprint
-        from onnx_diagnostic.torch_export_patch.eval import discover
+        from onnx_diagnostic.torch_export_patches.eval import discover
 
         pprint.pprint(discover())
     """