ColossalAI/colossalai/fx/tracer
Super Daniel d8a5aded19
[hotfix] change namespace for meta_trace. (#1541)
2022-09-06 11:46:12 +08:00
..
meta_patch [fx]patch nn.functional convolution (#1528) 2022-09-01 19:05:07 +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