| .. currentmodule:: torch.fx |
| |
| torch.fx |
| ============= |
| |
| Overview |
| -------- |
| .. automodule:: torch.fx |
| |
| Limitations of Symbolic Tracing |
| ------------------------------- |
| |
| TODO |
| |
| Writing Transformations |
| ----------------------- |
| |
| TODO |
| |
| Debugging Transformations |
| ------------------------- |
| |
| TODO |
| |
| API Reference |
| ------------- |
| |
| .. autofunction:: torch.fx.symbolic_trace |
| |
| .. autofunction:: torch.fx.wrap |
| |
| .. autoclass:: torch.fx.GraphModule |
| :members: |
| |
| .. automethod:: __init__ |
| |
| .. autoclass:: torch.fx.Graph |
| :members: |
| |
| .. automethod:: __init__ |
| |
| .. autoclass:: torch.fx.Node |
| :members: |
| |
| .. autoclass:: torch.fx.Tracer |
| :members: |
| |
| .. autoclass:: torch.fx.Proxy |
