Home
last modified time | relevance | path

Searched refs:NonDynamicallyQuantizableLinear (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantization_mappings.py98 nn.modules.linear.NonDynamicallyQuantizableLinear: nnq.Linear,
137 nn.modules.linear.NonDynamicallyQuantizableLinear: nnqat.Linear,
156 nn.modules.linear.NonDynamicallyQuantizableLinear: nnqd.Linear,
/aosp_15_r20/external/pytorch/torch/ao/ns/fx/
H A Dmappings.py362 (nn.Linear, nn.modules.linear.NonDynamicallyQuantizableLinear),
581 torch.nn.modules.linear.NonDynamicallyQuantizableLinear,
H A Dweight_utils.py195 nn.modules.linear.NonDynamicallyQuantizableLinear: mod_weight_detach,
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dlinear.py109 torch.nn.modules.linear.NonDynamicallyQuantizableLinear,
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dlinear.py136 class NonDynamicallyQuantizableLinear(Linear): class
H A Dactivation.py11 from .linear import NonDynamicallyQuantizableLinear
1097 self.out_proj = NonDynamicallyQuantizableLinear(
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dlinear.py149 _FLOAT_MODULE = (nn.Linear, nn.modules.linear.NonDynamicallyQuantizableLinear)
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_module.py233 … modules_under_test = [torch.nn.Linear, torch.nn.modules.linear.NonDynamicallyQuantizableLinear]
1673 … modules_under_test = [torch.nn.Linear, torch.nn.modules.linear.NonDynamicallyQuantizableLinear]