frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 21:49:24 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
17ab99463a62d3bde2cdbdd420b62784d15e5eef
pytorch/docs/source/export/api_reference.md
angelayi 66c9bc5062 [export] Add runnable code to export docs (#158506) 
Preview: https://docs-preview.pytorch.org/pytorch/pytorch/158506/export.html

Yay I can add runnable code to export docs now
Also moved export API reference to a different file.

With these changes, we can start to consolidate the [export tutorial](https://docs.pytorch.org/tutorials/intermediate/torch_export_tutorial.html) with the docs on pytorch docs. We just need to move the section on DDE and 0/1 specialization, and then I think we can delete the export tutorial.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158506
Approved by: https://github.com/pianpwk, https://github.com/svekars
2025-07-17 20:15:22 +00:00

1.5 KiB
Raw Blame History

(export.api_reference)=

torch.export API Reference

.. automodule:: torch.export

.. autofunction:: torch.export.export

.. autoclass:: torch.export.ExportedProgram
   :members:
   :exclude-members: __init__

.. automodule:: torch.export.dynamic_shapes
   :members: Dim, ShapesCollection, AdditionalInputs, refine_dynamic_shapes_from_suggested_fixes

.. autofunction:: torch.export.save

.. autofunction:: torch.export.load

.. autofunction:: torch.export.pt2_archive._package.package_pt2

.. autofunction:: torch.export.pt2_archive._package.load_pt2

.. autofunction:: torch.export.draft_export

.. automodule:: torch.export.unflatten
    :members:

.. autofunction:: torch.export.register_dataclass

.. automodule:: torch.export.decomp_utils
   :members:
   :ignore-module-all:
   :undoc-members:

.. automodule:: torch.export.experimental
   :members:
   :ignore-module-all:

.. automodule:: torch.export.passes
   :members:

.. automodule:: torch.export.pt2_archive
   :members:
   :ignore-module-all:

.. automodule:: torch.export.pt2_archive.constants
   :members:
   :ignore-module-all:

.. automodule:: torch.export.exported_program
   :members:
   :ignore-module-all:
   :exclude-members: ExportedProgram

.. automodule:: torch.export.custom_ops
   :members:
   :ignore-module-all:

.. automodule:: torch.export.custom_obj
   :members:
   :ignore-module-all:

.. automodule:: torch.export.graph_signature
   :members:
   :ignore-module-all:
   :undoc-members: