| # mypy: allow-untyped-defs | |
| __all__ = [ | |
| "get_static_sparse_quantized_mapping", | |
| "get_dynamic_sparse_quantized_mapping", | |
| ] | |
| def get_static_sparse_quantized_mapping(): | |
| import torch.ao.nn.sparse | |
| _static_sparse_quantized_mapping = { | |
| torch.nn.Linear: torch.ao.nn.sparse.quantized.Linear, | |
| } | |
| return _static_sparse_quantized_mapping | |
| def get_dynamic_sparse_quantized_mapping(): | |
| import torch.ao.nn.sparse | |
| _dynamic_sparse_quantized_mapping = { | |
| torch.nn.Linear: torch.ao.nn.sparse.quantized.dynamic.Linear, | |
| } | |
| return _dynamic_sparse_quantized_mapping | |