Searched full:tensor_qparams (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_workflow_ops.py | 338 …for float_type, torch_type, X, tensor_qparams in itertools.product(float_types, torch_types, Xs, t…
367 tensor_qparams = (True, False)
368 …oat_type, torch_type, tensor_qparam in itertools.product(float_types, torch_types, tensor_qparams):
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | QuantizedTensor.cpp | 187 TORCH_SELECTIVE_NAME("aten::quantize_per_tensor.tensor_qparams"), in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 482 - aten::quantize_per_tensor.tensor_qparams
929 aten::quantize_per_tensor.tensor_qparams: 1
|
| H A D | model_ops.yaml | 287 aten::quantize_per_tensor.tensor_qparams: 7
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 1038 aten::quantize_per_tensor.tensor_qparams
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1407 …"aten::quantize_per_tensor.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, ScalarType…
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | lowerings.cpp | 1927 …"aten::quantize_per_tensor.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, int dtype)… in nnc_lowerings_lazy_registration()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | serialized_shape_function_registry.cpp | 3315 …{"aten::quantize_per_tensor.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, ScalarTyp… in GetShapeFunctionMappings()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 7490 - func: quantize_per_tensor.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point, ScalarType…
7577 - func: fake_quantize_per_tensor_affine.tensor_qparams(Tensor self, Tensor scale, Tensor zero_point…
|