1# mypy: allow-untyped-defs 2__all__ = [ 3 "get_static_sparse_quantized_mapping", 4 "get_dynamic_sparse_quantized_mapping", 5] 6 7 8def get_static_sparse_quantized_mapping(): 9 import torch.ao.nn.sparse 10 11 _static_sparse_quantized_mapping = { 12 torch.nn.Linear: torch.ao.nn.sparse.quantized.Linear, 13 } 14 return _static_sparse_quantized_mapping 15 16 17def get_dynamic_sparse_quantized_mapping(): 18 import torch.ao.nn.sparse 19 20 _dynamic_sparse_quantized_mapping = { 21 torch.nn.Linear: torch.ao.nn.sparse.quantized.dynamic.Linear, 22 } 23 return _dynamic_sparse_quantized_mapping 24