| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_xnnpack_quantizer.py | 53 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
61 torch.ops.quantized_decomposed.quantize_per_tensor.default,
78 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
87 torch.ops.quantized_decomposed.quantize_per_tensor.default,
103 torch.ops.quantized_decomposed.quantize_per_tensor.default: 4,
111 torch.ops.quantized_decomposed.quantize_per_tensor.default,
114 torch.ops.quantized_decomposed.quantize_per_tensor.default,
137 torch.ops.quantized_decomposed.quantize_per_tensor.default: 3,
170 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
195 torch.ops.quantized_decomposed.quantize_per_tensor.default: 5,
[all …]
|
| H A D | test_x86inductor_quantizer.py | 596 torch.ops.quantized_decomposed.quantize_per_tensor.default: 1,
603 torch.ops.quantized_decomposed.quantize_per_tensor.default,
664 torch.ops.quantized_decomposed.quantize_per_tensor.default: 3,
671 torch.ops.quantized_decomposed.quantize_per_tensor.default,
702 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
714 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
721 torch.ops.quantized_decomposed.quantize_per_tensor.default,
752 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
759 torch.ops.quantized_decomposed.quantize_per_tensor.default,
762 torch.ops.quantized_decomposed.quantize_per_tensor.default,
[all …]
|
| H A D | test_quantize_pt2e.py | 113 torch.ops.quantized_decomposed.quantize_per_tensor.default: 2,
120 torch.ops.quantized_decomposed.quantize_per_tensor.default,
186 torch.ops.quantized_decomposed.quantize_per_tensor.default
280 torch.ops.quantized_decomposed.quantize_per_tensor.default
295 torch.ops.quantized_decomposed.quantize_per_tensor.default
384 torch.ops.quantized_decomposed.quantize_per_tensor.default
402 torch.ops.quantized_decomposed.quantize_per_tensor.default
484 torch.ops.quantized_decomposed.quantize_per_tensor.default
507 torch.ops.quantized_decomposed.quantize_per_tensor.default
564 == torch.ops.quantized_decomposed.quantize_per_tensor.default
[all …]
|
| H A D | test_metadata_porting.py | 48 torch.ops.quantized_decomposed.quantize_per_tensor.default,
50 torch.ops.quantized_decomposed.quantize_per_tensor.tensor,
189 torch.ops.quantized_decomposed.quantize_per_tensor.default: quantize_per_tensor_tags,
239 torch.ops.quantized_decomposed.quantize_per_tensor.default: quantize_per_tensor_tags,
308 torch.ops.quantized_decomposed.quantize_per_tensor.default: quantize_per_tensor_tags,
309 … torch.ops.quantized_decomposed.quantize_per_tensor.tensor: quantize_per_tensor_tensor_tags,
373 … torch.ops.quantized_decomposed.quantize_per_tensor.tensor: quantize_per_tensor_tensor_tags,
417 … torch.ops.quantized_decomposed.quantize_per_tensor.tensor: quantize_per_tensor_tensor_tags,
511 … torch.ops.quantized_decomposed.quantize_per_tensor.default: quantize_per_tensor_tensor_tags,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 430 auto in_cpu_quantized = at::quantize_per_tensor( in TEST_F()
468 auto in_q_cpu = at::quantize_per_tensor(in_cpu, scale, zero_point, dtype); in test_cpu_to_vulkan_and_vulkan_to_cpu()
590 auto in_q_cpu = at::quantize_per_tensor(in_cpu, scale, zero_point, dtype); in test_cpu_to_vulkan_and_dequantize()
709 const auto out_cpu = at::quantize_per_tensor( in TEST_F()
711 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F()
742 at::quantize_per_tensor(input, input_scale, input_zero_point, dtype); in test_quantize_per_tensor_and_vulkan_to_cpu()
744 at::Tensor out_q_vk = at::quantize_per_tensor( in test_quantize_per_tensor_and_vulkan_to_cpu()
886 const auto out_cpu = at::quantize_per_tensor( in TEST_F()
888 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F()
931 ? at::quantize_per_tensor( in test_quantize_per_tensor_and_dequantize()
[all …]
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A D | test_fusion_ops_passes.py | 224 x = torch.ops.quantized_decomposed.quantize_per_tensor(
245 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: 0,
257 x = torch.ops.quantized_decomposed.quantize_per_tensor(
276 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: 1,
288 x = torch.ops.quantized_decomposed.quantize_per_tensor(
308 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: 0,
324 x = torch.ops.quantized_decomposed.quantize_per_tensor(
338 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: 0,
354 x = torch.ops.quantized_decomposed.quantize_per_tensor(
368 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default: 0,
[all …]
|
| H A D | test_reorder_ops_passes.py | 65 x1 = exir_ops.edge.quantized_decomposed.quantize_per_tensor(
69 x2 = exir_ops.edge.quantized_decomposed.quantize_per_tensor(
73 x3 = exir_ops.edge.quantized_decomposed.quantize_per_tensor(
77 x4 = exir_ops.edge.quantized_decomposed.quantize_per_tensor(
92 nodes[0] == exir_ops.edge.quantized_decomposed.quantize_per_tensor
99 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default,
154 exir_ops.edge.cadence.quantize_per_tensor.default,
161 exir_ops.edge.cadence.quantize_per_tensor.default,
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 207 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
230 qY_hat = torch.quantize_per_tensor(dqY_hat, scale=output_scale,
417 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
424 qY = torch.quantize_per_tensor(dqY, scale=scale, zero_point=zero_point,
441 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
448 … qY_hat = torch.quantize_per_tensor(dqY_hat, scale=output_scale, zero_point=output_zero_point,
467 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
473 … qY_hat = torch.quantize_per_tensor(dqY_hat, scale=output_scale, zero_point=output_zero_point,
498 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
504 qY = torch.quantize_per_tensor(dqY, scale=scale, zero_point=zero_point,
[all …]
|
| H A D | test_quantized_tensor.py | 145 x_q = torch.quantize_per_tensor(x, 0.1, 10, torch.quint4x2)
146 y_q = torch.quantize_per_tensor(x, 0.1, 10, torch.quint4x2)
159 qx = torch.quantize_per_tensor(x, scale=scale, zero_point=zero_point, dtype=dtype)
170 qx = torch.quantize_per_tensor(x, scale=scale, zero_point=zero_point, dtype=dtype)
176 qx = torch.quantize_per_tensor(x, scale=scale, zero_point=zero_point, dtype=dtype)
222 q_s = torch.quantize_per_tensor(mat2quant, scale, zero_pt, dtype)
233 qr = torch.quantize_per_tensor(r, scale, zero_point, dtype)
252 empty_qr = torch.quantize_per_tensor(empty_r, scale, zero_point, dtype)
269 qr = torch.quantize_per_tensor(r, scale, zero_point, dtype)
292 qr = torch.quantize_per_tensor(r, scale, zero_point, dtype)
[all …]
|
| H A D | test_quantized_module.py | 61 qx = torch.quantize_per_tensor(x, 1.0, 0, dtype=torch.qint32)
116 W_q = torch.quantize_per_tensor(W, 0.1, W_zp, torch.qint8)
119 X_q = torch.quantize_per_tensor(X, 0.2, 10, torch.quint8)
255 qr = torch.quantize_per_tensor(r, scale, zero_point, dtype)
316 Y_exp = torch.quantize_per_tensor(
910 qX = torch.quantize_per_tensor(X, scale=scale, zero_point=zero_point,
930 quant_ref = torch.quantize_per_tensor(y_ref, 1.0, 0, dtype=torch.quint8)
933 qx = torch.quantize_per_tensor(x, 1.0, 0, dtype=torch.quint8)
981 quant_ref = torch.quantize_per_tensor(y_ref, 1.0, 0, dtype=torch.quint8)
984 qx = torch.quantize_per_tensor(x, 1.0, 0, dtype=torch.quint8)
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_quantization.cpp | 45 auto q = at::quantize_per_tensor(x, 0.1f, 13, at::kQInt8); in TEST_F()
75 auto q = at::quantize_per_tensor(x, 0.1f, 122, at::kQUInt8); in TEST_F()
107 auto q = at::quantize_per_tensor(x, 0.1f, 122, at::kQUInt8); in TEST_F()
157 auto q1 = at::quantize_per_tensor(x1, 0.1f, 13, at::kQInt8); in TEST_F()
158 auto q2 = at::quantize_per_tensor(x2, 0.1f, 13, at::kQInt8); in TEST_F()
200 auto q1 = at::quantize_per_tensor(x1, 0.1f, 13, at::kQUInt8); in TEST_F()
201 auto q2 = at::quantize_per_tensor(x2, 0.1f, 13, at::kQUInt8); in TEST_F()
238 auto q1 = at::quantize_per_tensor(x1, 0.1f, 13, at::kQUInt8); in TEST_F()
292 auto q1 = at::quantize_per_tensor(x1, 0.1f, 13, at::kQUInt8); in TEST_F()
293 auto q2 = at::quantize_per_tensor(x2, 0.1f, 13, at::kQUInt8); in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 619 ns.call_function(torch.quantize_per_tensor),
626 ns.call_function(torch.quantize_per_tensor),
631 ns.call_function(torch.quantize_per_tensor),
638 ns.call_function(torch.quantize_per_tensor),
646 ns.call_function(torch.quantize_per_tensor),
653 ns.call_function(torch.quantize_per_tensor),
658 ns.call_function(torch.quantize_per_tensor),
665 ns.call_function(torch.quantize_per_tensor),
673 ns.call_function(torch.quantize_per_tensor),
680 ns.call_function(torch.quantize_per_tensor),
[all …]
|
| H A D | test_quantize_fx.py | 538 ns.call_function(torch.quantize_per_tensor): 1 if two_conv else 2,
545 ns.call_function(torch.quantize_per_tensor): 3,
638 ns.call_function(torch.quantize_per_tensor),
644 ns.call_function(torch.quantize_per_tensor),
2000 ns.call_function(torch.quantize_per_tensor) : 1,
2007 ns.call_function(torch.quantize_per_tensor) : 1,
2034 ns.call_function(torch.quantize_per_tensor) : 1,
2042 ns.call_function(torch.quantize_per_tensor) : 0,
2077 ns.call_function(torch.quantize_per_tensor),
2106 ns.call_function(torch.quantize_per_tensor),
[all …]
|
| /aosp_15_r20/external/executorch/backends/transforms/test/ |
| H A D | test_duplicate_dynamic_quant_chain.py | 97 torch.ops.quantized_decomposed.quantize_per_tensor.tensor: 2,
129 torch.ops.quantized_decomposed.quantize_per_tensor.tensor: 1,
137 torch.ops.quantized_decomposed.quantize_per_tensor.tensor: 2,
167 torch.ops.quantized_decomposed.quantize_per_tensor.tensor: 1,
175 torch.ops.quantized_decomposed.quantize_per_tensor.tensor: 1,
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | utils.py | 25 torch.ops.quantized_decomposed.quantize_per_tensor.default,
26 torch.ops.quantized_decomposed.quantize_per_tensor.tensor,
404 …torch.ops.quantized_decomposed.quantize_per_tensor.default: torch.ops.quantized_decomposed.quantiz…
406 …torch.ops.quantized_decomposed.quantize_per_tensor.tensor: torch.ops.quantized_decomposed.quantize…
408 …torch.ops.quantized_decomposed.quantize_per_tensor.tensor2: torch.ops.quantized_decomposed.quantiz…
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_ondevice_quantization.py | 194 quantize_per_tensor = quantize_per_channel = 0
197 quantize_per_tensor += 1
200 self.assertEqual(quantize_per_tensor + quantize_per_channel, num_nodes)
261 quantize_per_tensor = quantize_per_channel = 0
276 quantize_per_tensor += 1
277 self.assertEqual(quantize_per_tensor + quantize_per_channel, num_nodes)
318 quantize_per_tensor = quantize_per_channel = 0
324 quantize_per_tensor += 1
336 self.assertEqual(quantize_per_tensor, 0)
|
| /aosp_15_r20/external/executorch/kernels/quantized/test/ |
| H A D | test_out_variants.py | 77 self.assertIsNotNone(ops.edge.quantized_decomposed.quantize_per_tensor.out)
78 fn = ops.edge.quantized_decomposed.quantize_per_tensor.default
86 ops.edge.quantized_decomposed.quantize_per_tensor.Tensor_out
88 fn = ops.edge.quantized_decomposed.quantize_per_tensor.tensor
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | qconv_test.py | 22 qX = torch.quantize_per_tensor(
27 self.qW = torch.quantize_per_tensor(
51 qX = torch.quantize_per_tensor(
56 self.qW = torch.quantize_per_tensor(
|
| /aosp_15_r20/external/executorch/kernels/quantized/ |
| H A D | targets.bzl | 27 "quantized_decomposed::quantize_per_tensor.out",
28 "quantized_decomposed::quantize_per_tensor.Tensor_out",
70 "quantized_decomposed::quantize_per_tensor.out",
71 "quantized_decomposed::quantize_per_tensor.Tensor_out",
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | reorder_ops.py | 115 torch.ops.quantized_decomposed.quantize_per_tensor,
116 exir_ops.edge.quantized_decomposed.quantize_per_tensor,
186 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default
208 exir_ops.edge.quantized_decomposed.quantize_per_tensor.default
297 exir_ops.edge.quantized_decomposed.quantize_per_tensor,
298 torch.ops.quantized_decomposed.quantize_per_tensor,
410 exir_ops.edge.quantized_decomposed.quantize_per_tensor,
556 exir_ops.edge.quantized_decomposed.quantize_per_tensor,
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | activation.py | 262 bias_k = torch.quantize_per_tensor(bias_k, sc, zp, torch.quint8)
270 bias_v = torch.quantize_per_tensor(bias_v, sc, zp, torch.quint8)
296 qw = torch.quantize_per_tensor(w, scale=1.0, zero_point=0, dtype=torch.quint8)
322 qweight = torch.quantize_per_tensor(
339 qweight = torch.quantize_per_tensor(
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | quantization_ops.py | 18 a = torch.quantize_per_tensor(torch.tensor([3.0]), 1.0, 0, torch.qint32)
19 b = torch.quantize_per_tensor(torch.tensor(4.0), 1.0, 0, torch.qint32)
20 c = torch.quantize_per_tensor(
35 torch.quantize_per_tensor(
40 torch.quantize_per_tensor(
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/passes/ |
| H A D | test_tag_implicit_q_dq_pass.py | 29 x = exir_ops.edge.quantized_decomposed.quantize_per_tensor.default(
36 x = exir_ops.edge.quantized_decomposed.quantize_per_tensor.default(
43 x = exir_ops.edge.quantized_decomposed.quantize_per_tensor.default(
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | README.md | 335 …quantize_per_tensor = torch.quantize_per_tensor(x, linear_input_scale_0, linear_input_zero_point_0…
336 dequantize = quantize_per_tensor.dequantize(); quantize_per_tensor = None
340 …quantize_per_tensor_1 = torch.quantize_per_tensor(linear, linear_scale_0, linear_zero_point_0, tor…
382 The quantize operator we can choose from right now are: (quantize_per_tensor, quantize_per_channel,…
413 …quantize_per_tensor = torch.quantize_per_tensor(x, linear_input_scale_0, linear_input_zero_point_0…
414 linear = self.linear(quantize_per_tensor); quantize_per_tensor = None
426 dequantize -> float_op -> quantize_per_tensor
437 weight - quantize_per_tensor - dequantize /
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_kernels.py | 75 X_q = torch.quantize_per_tensor(
95 W_q = torch.quantize_per_tensor(
157 X_q = torch.quantize_per_tensor(
162 W_q = torch.quantize_per_tensor(W_fp32, W_scale, W_zp, torch.qint8)
|