├── .flake8 ├── .github └── workflows │ ├── main.yaml │ └── publish.yaml ├── .gitignore ├── .isort.cfg ├── CHANGELOG.md ├── LICENSE ├── README.md ├── ptflops ├── __init__.py ├── aten_engine.py ├── aten_ops.py ├── flops_counter.py ├── pytorch_engine.py ├── pytorch_ops.py └── utils.py ├── pyproject.toml ├── samples ├── bert.py └── classification.py └── tests └── common_test.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/.github/workflows/main.yaml -------------------------------------------------------------------------------- /.github/workflows/publish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/.github/workflows/publish.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/.isort.cfg -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /ptflops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/__init__.py -------------------------------------------------------------------------------- /ptflops/aten_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/aten_engine.py -------------------------------------------------------------------------------- /ptflops/aten_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/aten_ops.py -------------------------------------------------------------------------------- /ptflops/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/flops_counter.py -------------------------------------------------------------------------------- /ptflops/pytorch_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/pytorch_engine.py -------------------------------------------------------------------------------- /ptflops/pytorch_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/pytorch_ops.py -------------------------------------------------------------------------------- /ptflops/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/ptflops/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /samples/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/samples/bert.py -------------------------------------------------------------------------------- /samples/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/samples/classification.py -------------------------------------------------------------------------------- /tests/common_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sovrasov/flops-counter.pytorch/HEAD/tests/common_test.py --------------------------------------------------------------------------------