ColossalAI/colossalai/auto_parallel/meta_profiler/meta_registry
YuliangLiu0306 1cce6e36ca
[autoparallel] use metainfo in handler (#2149)
2022-12-20 10:31:22 +08:00
..
__init__.py [autoparallel] add binary elementwise metainfo for auto parallel (#2058) 2022-12-04 15:18:51 +08:00
activation.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
binary_elementwise_ops.py [autoparallel] add binary elementwise metainfo for auto parallel (#2058) 2022-12-04 15:18:51 +08:00
conv.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
linear.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
norm.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
pooling.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00