ColossalAI/colossalai/fx/tracer
Xue Fuzhao e070ca45c6 [NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563) 2022-09-08 22:11:04 +08:00
..
meta_patch [NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563) 2022-09-08 22:11:04 +08:00
__init__.py [fx] support meta tracing for aten level computation graphs like functorch. (#1536) 2022-09-05 12:10:09 +08:00
_meta_trace.py [hotfix] change namespace for meta_trace. (#1541) 2022-09-06 11:46:12 +08:00
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
tracer.py [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00