Home
last modified time | relevance | path

Searched defs:qscheme (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/
H A Dutils.py188 weight_qscheme: torch.qscheme,
249 weight_qscheme: torch.qscheme,
304 weight_qscheme: torch.qscheme,
334 weight_qscheme: torch.qscheme,
377 weight_qscheme: torch.qscheme,
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_module.py74 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
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dobserver.py216 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
H A Dfake_quantize.py50 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
H A Dutils.py155 def is_per_tensor(qscheme): argument
159 def is_per_channel(qscheme): argument
623 qscheme: torch.qscheme = torch.per_tensor_affine, argument
H A D__init__.py198 qscheme: Optional[torch.qscheme] = None, argument
/aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/
H A Dquantizer.py45 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
/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.h89 QScheme qscheme() const override { in qscheme() function
140 QScheme qscheme() const override { in qscheme() function
203 QScheme qscheme() const override { in qscheme() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp30 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
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_qschemes.cpp24 auto qscheme = static_cast<at::QScheme>(i); in initializeQSchemes() local
35 PyObject* getTHPQScheme(at::QScheme qscheme) { in getTHPQScheme()
/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dreplace_quantized_ops_with_standard_ops_pass.py51 qscheme: Optional[torch.qscheme], argument
74 qscheme: Optional[torch.qscheme], argument
108 qscheme: Optional[torch.qscheme], argument
/aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/
H A Dqobserver_test.py107 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
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.cpp137 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
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_equalize.py66 qscheme=torch.per_tensor_affine, argument
167 qscheme=torch.per_tensor_affine, argument
/aosp_15_r20/external/pytorch/torch/csrc/
H A DQScheme.h12 PyObject_HEAD at::QScheme qscheme; member
H A DQScheme.cpp13 PyObject* THPQScheme_New(at::QScheme qscheme, const std::string& name) { in THPQScheme_New()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DTensorOperators.cpp92 auto qscheme = self.quantizer()->qscheme(); in AT_FORALL_OPERATORS() local
H A DSorting.cpp51 auto qscheme = self.qscheme(); in topk_quantized_cpu() local
/aosp_15_r20/external/pytorch/c10/core/
H A DQScheme.h33 inline std::string toString(QScheme qscheme) { in toString()
/aosp_15_r20/external/pytorch/torch/ao/quantization/experimental/
H A Dadaround_fake_quantize.py27 qscheme=torch.per_tensor_symmetric, # not used, but needed for fakequant argument
H A Dadaround_optimization.py35 qscheme: torch.qscheme = torch.per_tensor_symmetric, argument
/aosp_15_r20/external/executorch/backends/qualcomm/quantizer/observers/
H A Dper_channel_param_observer.py17 qscheme=torch.per_channel_symmetric, argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/utils/
H A Dwrap_outputs.h77 inline PyObject* wrap(at::QScheme qscheme) { in wrap()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dunpickler.cpp841 std::optional<at::QScheme> qscheme; in readGlobal() local
951 auto qscheme = static_cast<at::QScheme>(qparams.at(0).toInt()); in rebuildTensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPadNd.cpp84 const auto qscheme = self.qscheme(); in constant_pad_nd() local

12