ColossalAI/colossalai/fx/passes/experimental
YuliangLiu0306 51b89d2202
[autoparallel] runtime_backward_apply (#1720)
2022-10-18 10:44:58 +08:00
..
adding_shape_consistency_pass.py [fx] PoC of runtime shape consistency application (#1607) 2022-09-20 14:00:04 +08:00
adding_shape_consistency_pass_v2.py [autoparallel] runtime_backward_apply (#1720) 2022-10-18 10:44:58 +08:00