| /external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| D | utils.py | 188 weight_qscheme: torch.qscheme,
249 weight_qscheme: torch.qscheme,
304 weight_qscheme: torch.qscheme,
334 weight_qscheme: torch.qscheme,
377 weight_qscheme: torch.qscheme,
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_qschemes.cpp | 24 auto qscheme = static_cast<at::QScheme>(i); in initializeQSchemes() local
35 PyObject* getTHPQScheme(at::QScheme qscheme) { in getTHPQScheme()
|
| /external/pytorch/torch/ao/quantization/quantizer/ |
| D | quantizer.py | 45 qscheme: Optional[torch.qscheme] = None variable in QuantizationSpec
74 qscheme: Optional[torch.qscheme] = None variable in FixedQParamsQuantizationSpec
107 qscheme: Optional[torch.qscheme] = None variable in DerivedQuantizationSpec
|
| /external/pytorch/aten/src/ATen/quantized/ |
| D | Quantizer.h | 89 QScheme qscheme() const override { in qscheme() function
140 QScheme qscheme() const override { in qscheme() function
203 QScheme qscheme() const override { in qscheme() function
|
| /external/pytorch/test/quantization/core/ |
| D | test_workflow_module.py | 74 def test_per_tensor_observers(self, qdtype, qscheme, reduce_range): argument
84 def _get_ref_params(reduce_range, qscheme, dtype, input_scale, min_val, max_val): argument
156 def test_per_channel_observers(self, qdtype, qscheme, ch_axis, reduce_range): argument
657 def test_observer_scriptable(self, qdtype, qscheme): argument
679 def test_histogram_observer(self, qdtype, qscheme, reduce_range): argument
761 def test_histogram_observer_against_reference(self, N, bins, dtype, qscheme, reduce_range): argument
|
| /external/pytorch/torch/ao/quantization/ |
| D | fake_quantize.py | 50 def _is_per_channel(qscheme: "torch.qscheme") -> bool: argument
58 def _is_per_tensor(qscheme: "torch.qscheme") -> bool: argument
62 def _is_symmetric_quant(qscheme: "torch.qscheme") -> bool: argument
66 def _is_float_qparams(qscheme: "torch.qscheme") -> bool: argument
|
| D | utils.py | 155 def is_per_tensor(qscheme): argument
159 def is_per_channel(qscheme): argument
623 qscheme: torch.qscheme = torch.per_tensor_affine, argument
|
| D | observer.py | 216 qscheme=torch.per_tensor_affine, argument
482 qscheme=torch.per_tensor_affine, argument
609 qscheme=torch.per_tensor_affine, argument
692 qscheme=torch.per_channel_affine, argument
899 qscheme=torch.per_channel_affine, argument
988 qscheme=torch.per_tensor_affine, argument
1401 qscheme=torch.per_tensor_affine, argument
1457 qscheme=None, argument
|
| D | __init__.py | 198 qscheme: Optional[torch.qscheme] = None, argument
|
| /external/pytorch/benchmarks/operator_benchmark/pt/ |
| D | qobserver_test.py | 107 def init(self, C, M, N, dtype, qscheme, op_func, device): argument
117 def init(self, C, M, N, dtype, qscheme, op_func, device): argument
|
| /external/pytorch/torch/csrc/ |
| D | QScheme.h | 12 PyObject_HEAD at::QScheme qscheme; member
|
| D | QScheme.cpp | 13 PyObject* THPQScheme_New(at::QScheme qscheme, const std::string& name) { in THPQScheme_New()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | TensorOperators.cpp | 92 auto qscheme = self.quantizer()->qscheme(); in AT_FORALL_OPERATORS() local
|
| D | Sorting.cpp | 51 auto qscheme = self.qscheme(); in topk_quantized_cpu() local
|
| /external/pytorch/c10/core/ |
| D | QScheme.h | 33 inline std::string toString(QScheme qscheme) { in toString()
|
| /external/pytorch/torch/_export/passes/ |
| D | replace_quantized_ops_with_standard_ops_pass.py | 51 qscheme: Optional[torch.qscheme], argument
74 qscheme: Optional[torch.qscheme], argument
108 qscheme: Optional[torch.qscheme], argument
|
| /external/pytorch/torch/ao/quantization/experimental/ |
| D | adaround_fake_quantize.py | 27 qscheme=torch.per_tensor_symmetric, # not used, but needed for fakequant argument
|
| D | adaround_optimization.py | 35 qscheme: torch.qscheme = torch.per_tensor_symmetric, argument
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | insert_quant_dequant.cpp | 30 c10::QScheme qscheme{c10::kPerTensorAffine}; member
39 c10::QScheme toAffine(c10::QScheme qscheme) { in toAffine()
52 bool isPerChannel(at::QScheme qscheme) { in isPerChannel()
735 void checkQScheme(Graph* g, c10::QScheme qscheme) { in checkQScheme()
1030 auto qscheme = std::get<0>(tp); in quantizeTensors() local
1067 c10::QScheme qscheme = c10::kPerTensorAffine; in getQSchemeAndQParamVector() local
1174 auto qscheme = std::get<0>(*qparams_opt); in propagateQParams() local
1587 auto qscheme = observer_module.attr("qscheme").toQScheme(); in insertCalculateQParams() local
|
| /external/executorch/backends/qualcomm/quantizer/observers/ |
| D | per_channel_param_observer.py | 17 qscheme=torch.per_channel_symmetric, argument
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | pybind_utils.cpp | 137 auto qscheme = reinterpret_cast<THPQScheme*>(obj.ptr()); in toIValue() local
146 auto qscheme = reinterpret_cast<THPQScheme*>(obj.ptr()); in toIValue() local
492 auto qscheme = reinterpret_cast<THPQScheme*>(obj.ptr()); in toIValue() local
|
| /external/pytorch/torch/csrc/autograd/utils/ |
| D | wrap_outputs.h | 77 inline PyObject* wrap(at::QScheme qscheme) { in wrap()
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | _equalize.py | 66 qscheme=torch.per_tensor_affine, argument
167 qscheme=torch.per_tensor_affine, argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | PadNd.cpp | 84 const auto qscheme = self.qscheme(); in constant_pad_nd() local
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantized.py | 51 def _calculate_dynamic_qparams(X, dtype, reduce_range=False, qscheme=torch.per_tensor_affine): argument
|