├── .gitignore ├── LICENSE ├── README.md ├── examples ├── profile_torchvision.py ├── profile_transformer.py └── trace_linear.py ├── setup.py └── torchprofile ├── __init__.py ├── handlers.py ├── profile.py ├── utils ├── __init__.py ├── flatten.py ├── ir │ ├── __init__.py │ ├── graph.py │ ├── node.py │ └── variable.py ├── math.py └── trace.py └── version.py /.gitignore: -------------------------------------------------------------------------------- 1 | .vscode/ 2 | __pycache__/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/README.md -------------------------------------------------------------------------------- /examples/profile_torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/examples/profile_torchvision.py -------------------------------------------------------------------------------- /examples/profile_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/examples/profile_transformer.py -------------------------------------------------------------------------------- /examples/trace_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/examples/trace_linear.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/setup.py -------------------------------------------------------------------------------- /torchprofile/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/__init__.py -------------------------------------------------------------------------------- /torchprofile/handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/handlers.py -------------------------------------------------------------------------------- /torchprofile/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/profile.py -------------------------------------------------------------------------------- /torchprofile/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchprofile/utils/flatten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/flatten.py -------------------------------------------------------------------------------- /torchprofile/utils/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/ir/__init__.py -------------------------------------------------------------------------------- /torchprofile/utils/ir/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/ir/graph.py -------------------------------------------------------------------------------- /torchprofile/utils/ir/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/ir/node.py -------------------------------------------------------------------------------- /torchprofile/utils/ir/variable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/ir/variable.py -------------------------------------------------------------------------------- /torchprofile/utils/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/math.py -------------------------------------------------------------------------------- /torchprofile/utils/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijian-liu/torchprofile/HEAD/torchprofile/utils/trace.py -------------------------------------------------------------------------------- /torchprofile/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.0.4' 2 | --------------------------------------------------------------------------------