Home
last modified time | relevance | path

Searched full:_make_per_tensor_quantized_tensor (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_tensor.py479 q = torch._make_per_tensor_quantized_tensor(int_tensor, scale, zero_point)
904 q = torch._make_per_tensor_quantized_tensor(q_int, scale=scale, zero_point=zero_point)
1193 q = torch._make_per_tensor_quantized_tensor(q_int, scale=scale, zero_point=zero_point)
1200 a = torch._make_per_tensor_quantized_tensor(a_int, scale=scale, zero_point=zero_point)
1215 a = torch._make_per_tensor_quantized_tensor(a_int, scale=scale, zero_point=zero_point)
1233 q1 = torch._make_per_tensor_quantized_tensor(q1_int, scale=scale, zero_point=zero_point)
1246 a = torch._make_per_tensor_quantized_tensor(a_int, scale=scale, zero_point=zero_point)
1262 q1 = torch._make_per_tensor_quantized_tensor(a_int, scale=scale, zero_point=zero_point)
1274 q = torch._make_per_tensor_quantized_tensor(q_int, scale=scale, zero_point=zero_point)
1282 a = torch._make_per_tensor_quantized_tensor(a_int, scale=scale, zero_point=zero_point)
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorCompare.cpp29 #include <ATen/ops/_make_per_tensor_quantized_tensor.h>
672 at::_make_per_tensor_quantized_tensor(max, self.q_scale(), self.q_zero_point()), max_indices); in qmax()
682 at::_make_per_tensor_quantized_tensor(min, self.q_scale(), self.q_zero_point()), min_indices); in qmin()
H A Dnative_functions.yaml7556 - func: _make_per_tensor_quantized_tensor(Tensor self, float scale, int zero_point) -> Tensor
7560 autogen: _make_per_tensor_quantized_tensor.out
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DActivation.cpp26 return at::_make_per_tensor_quantized_tensor(relu_int_repr, self.q_scale(), zero_point); in relu_quantized_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorCompare.cpp56 at::_make_per_tensor_quantized_tensor( in sort_quantized_cpu_stable()
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dutils.py53 qweight = torch._make_per_tensor_quantized_tensor(
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect417 aten::_make_per_tensor_quantized_tensor
418 aten::_make_per_tensor_quantized_tensor.out
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmodel_ops.yaml37 aten::_make_per_tensor_quantized_tensor: 1
H A Dcoverage.yaml1054 aten::_make_per_tensor_quantized_tensor: 2
/aosp_15_r20/external/pytorch/torch/backends/_nnapi/
H A Dserializer.py1852 unsigned_weight = torch._make_per_tensor_quantized_tensor(
2033 unsigned_weight = torch._make_per_tensor_quantized_tensor(
/aosp_15_r20/external/pytorch/test/
H A Dallowlist_for_publicAPI.json1266 "_make_per_tensor_quantized_tensor",
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py1514 "torch._make_per_tensor_quantized_tensor",