ColossalAI/colossalai/fx/passes/experimental
YuliangLiu0306 d2fc067231
[autoparallel] fix param hook issue in transform pass (#1755)
2022-10-24 13:13:38 +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] fix param hook issue in transform pass (#1755) 2022-10-24 13:13:38 +08:00