ColossalAI/colossalai/_analyzer/fx/passes
..
__init__.py
graph_profile.py
shape_prop.py