ColossalAI/colossalai/fx/tracer
YuliangLiu0306 d87baa85d9
[autoparallel] support linear function bias addition (#2104)
2022-12-09 10:31:36 +08:00
..
bias_addition_patch [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00
meta_patch
__init__.py
_meta_trace.py
_symbolic_trace.py
_tracer_utils.py
experimental.py
registry.py [autoparallel] support addbmm computation (#2102) 2022-12-08 21:15:11 +08:00
tracer.py [autoparallel] support linear function bias addition (#2104) 2022-12-09 10:31:36 +08:00