.. py:module:: megengine.traced_module
.. currentmodule:: megengine.traced_module

=======================
megengine.traced_module
=======================

>>> import megengine.traced_module as tm

.. autosummary::
   :toctree: api
   :nosignatures:

   TracedModule

.. autosummary::
   :toctree: api
   :nosignatures:

   trace_module
   register_as_builtin
   wrap

.. py:module:: megengine.traced_module.node
.. currentmodule:: megengine.traced_module.node

Node
~~~~
.. autosummary::
   :toctree: api
   :nosignatures:

   ModuleNode
   TensorNode

.. py:module:: megengine.traced_module.expr
.. currentmodule:: megengine.traced_module.expr

Expr
~~~~
.. autosummary::
   :toctree: api
   :nosignatures:

   Input
   GetAttr
   CallMethod
   CallFunction
   Constant
   Apply

.. py:module:: megengine.traced_module.traced_module
.. currentmodule:: megengine.traced_module.traced_module

InternalGraph
-------------
.. autosummary::
   :toctree: api
   :nosignatures:

   InternalGraph