mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 12:26:07 +00:00
37 lines
564 B
ReStructuredText
37 lines
564 B
ReStructuredText
``jax.export`` module
|
|
=====================
|
|
|
|
.. automodule:: jax.export
|
|
|
|
Classes
|
|
-------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Exported
|
|
DisabledSafetyCheck
|
|
|
|
Functions
|
|
---------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
export
|
|
deserialize
|
|
minimum_supported_serialization_version
|
|
maximum_supported_serialization_version
|
|
default_lowering_platform
|
|
|
|
Functions related to shape polymorphism
|
|
---------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
symbolic_shape
|
|
symbolic_args_specs
|
|
is_symbolic_dim
|
|
SymbolicScope
|