ColossalAI/colossalai/auto_parallel/tensor_shard/node_handler/experimental
YuliangLiu0306 81330b0352
[autoparallel] add experimental permute handler (#2029)
2022-11-27 20:26:52 +08:00
..
__init__.py [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00
permute_handler.py [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00
reshape_generator.py [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00
transpose_handler.py [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00
view_handler.py [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00