├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── conftest.py ├── examples ├── heuristic.py └── simple.py ├── paper ├── figures.ipynb ├── main.py └── utils.py ├── pyproject.toml ├── src └── upscale │ ├── __init__.py │ ├── masking │ ├── importance.py │ ├── mask.py │ └── utils.py │ └── pruning │ ├── README.md │ ├── __init__.py │ ├── manager.py │ ├── mock.py │ ├── pruner.py │ ├── reorder.py │ ├── resolve.py │ ├── trace.py │ ├── tracing.py │ └── utils.py └── tests ├── test_api.py └── test_core.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/README.md -------------------------------------------------------------------------------- /conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/conftest.py -------------------------------------------------------------------------------- /examples/heuristic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/examples/heuristic.py -------------------------------------------------------------------------------- /examples/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/examples/simple.py -------------------------------------------------------------------------------- /paper/figures.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/paper/figures.ipynb -------------------------------------------------------------------------------- /paper/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/paper/main.py -------------------------------------------------------------------------------- /paper/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/paper/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/upscale/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/__init__.py -------------------------------------------------------------------------------- /src/upscale/masking/importance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/masking/importance.py -------------------------------------------------------------------------------- /src/upscale/masking/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/masking/mask.py -------------------------------------------------------------------------------- /src/upscale/masking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/masking/utils.py -------------------------------------------------------------------------------- /src/upscale/pruning/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/README.md -------------------------------------------------------------------------------- /src/upscale/pruning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/__init__.py -------------------------------------------------------------------------------- /src/upscale/pruning/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/manager.py -------------------------------------------------------------------------------- /src/upscale/pruning/mock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/mock.py -------------------------------------------------------------------------------- /src/upscale/pruning/pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/pruner.py -------------------------------------------------------------------------------- /src/upscale/pruning/reorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/reorder.py -------------------------------------------------------------------------------- /src/upscale/pruning/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/resolve.py -------------------------------------------------------------------------------- /src/upscale/pruning/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/trace.py -------------------------------------------------------------------------------- /src/upscale/pruning/tracing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/tracing.py -------------------------------------------------------------------------------- /src/upscale/pruning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/src/upscale/pruning/utils.py -------------------------------------------------------------------------------- /tests/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/tests/test_api.py -------------------------------------------------------------------------------- /tests/test_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-upscale/HEAD/tests/test_core.py --------------------------------------------------------------------------------