Searched defs:qdtype (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/quantization/core/ |
| D | test_workflow_module.py | 74 def test_per_tensor_observers(self, qdtype, qscheme, reduce_range): argument
156 def test_per_channel_observers(self, qdtype, qscheme, ch_axis, reduce_range): argument
638 def test_observer_scriptable(self, qdtype): argument
657 def test_observer_scriptable(self, qdtype, qscheme): argument
679 def test_histogram_observer(self, qdtype, qscheme, reduce_range): argument
|
| /external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| D | quantization.cpp | 155 const auto dtype = [](auto qdtype) { in computeQuantizePerTensor()
230 const auto qdtype = std::get<int64_t>(inputs[3]); in computeQuantizePerTensorExternalCall() local
232 const auto dtype = [](auto qdtype) { in computeQuantizePerTensorExternalCall()
265 const int64_t qdtype = (int64_t)immQDType(qx); in computeDequantizeExternalCall() local
|
| /external/pytorch/torch/ao/quantization/ |
| D | utils.py | 174 def to_underlying_dtype(qdtype): argument
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | external_functions.cpp | 947 const c10::ScalarType qdtype = in nnc_aten_quantized_cat() local
1070 const c10::ScalarType qdtype = static_cast<c10::ScalarType>(extra_args[2]); in nnc_aten_quantize_per_tensor() local
1098 const c10::ScalarType qdtype = static_cast<c10::ScalarType>(extra_args[2]); in nnc_aten_quantize_per_tensor_out() local
1116 const int64_t qdtype = extra_args[2]; in nnc_aten_dequantize() local
1143 const int64_t qdtype = extra_args[2]; in nnc_aten_dequantize_out() local
|