Home
last modified time | relevance | path

Searched refs:WeightedQuantizedModule (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/experimental/
H A Dlinear.py5 from torch.ao.nn.quantized.modules.utils import WeightedQuantizedModule
10 class LinearAPoT(WeightedQuantizedModule):
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dlinear.py13 from .utils import _hide_packed_params_repr, _quantize_weight, WeightedQuantizedModule
120 class Linear(WeightedQuantizedModule):
H A Dutils.py15 class WeightedQuantizedModule(torch.nn.Module, metaclass=abc.ABCMeta): class
H A Dconv.py16 from .utils import _quantize_weight, WeightedQuantizedModule
40 class _ConvNd(WeightedQuantizedModule):
/aosp_15_r20/external/pytorch/torch/nn/quantized/modules/
H A Dutils.py16 WeightedQuantizedModule,
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_lower_to_native_backend.py14 from torch.ao.nn.quantized.modules.utils import WeightedQuantizedModule
249 STATIC_LOWER_MODULE_MAP: Dict[Type[nn.Module], Type[WeightedQuantizedModule]] = {
299 Type[nn.Module], Tuple[Type[nn.Module], Type[WeightedQuantizedModule]]
318 Type[nn.Module], Tuple[Type[nn.Module], Type[WeightedQuantizedModule]]