/usr/local/lib64/python3.6/site-packages/torch/ao/sparsity
Edit: /usr/local/lib64/python3.6/site-packages/torch/ao/sparsity/__init__.py (918B)
# Variables
from ._mappings import get_dynamic_sparse_quantized_mapping
from ._mappings import get_static_sparse_quantized_mapping
# Sparsifier
from .sparsifier.base_sparsifier import BaseSparsifier
from .sparsifier.weight_norm_sparsifier import WeightNormSparsifier
# Scheduler
from .scheduler.base_scheduler import BaseScheduler
from .scheduler.lambda_scheduler import LambdaSL
# Parametrizations
from .sparsifier.utils import FakeSparsity
from .sparsifier.utils import module_to_fqn
from .sparsifier.utils import fqn_to_module
# === Experimental ===
# Parametrizations
from .experimental.pruner.parametrization import PruningParametrization
from .experimental.pruner.parametrization import ZeroesParametrization
from .experimental.pruner.parametrization import ActivationReconstruction
from .experimental.pruner.parametrization import BiasHook
# Pruner
from .experimental.pruner.base_pruner import BasePruner