| /aosp_15_r20/external/executorch/kernels/quantized/test/ |
| H A D | op_embedding_test.cpp | 55 Tensor weight_zero_points = tf.full({3}, zero_point); in test_dtype() local
114 Tensor weight_zero_points = tf.full({3}, zero_point); in TEST() local
183 Tensor weight_zero_points = tf.make({3}, {1, 5, 7}); in TEST() local
240 Tensor weight_zero_points = tf.make({4}, {1, 5, 7, 5}); in TEST() local
270 Tensor weight_zero_points = tf.make({2}, {1, 5}); in TEST() local
300 Tensor weight_zero_points = tf.make({3, 2}, {1, 5, 7, 9, 11, 13}); in TEST() local
329 Tensor weight_zero_points = tf.make({3}, {1, 5, 7}); in TEST() local
358 Tensor weight_zero_points = tf.make({3, 3}, {1, 5, 7, 1, 5, 7, 1, 5, 7}); in TEST() local
|
| H A D | op_embedding2b_test.cpp | 39 Tensor weight_zero_points = tf.make({3}, {1, -2, 0}); in TEST() local
117 Tensor weight_zero_points = tf.make({4}, {1, -2, 1, 0}); in TEST() local
145 Tensor weight_zero_points = tf.make({2}, {1, -2}); in TEST() local
173 Tensor weight_zero_points = tf.make({2, 3}, {0, 0, 0, 0, 0, 0}); in TEST() local
|
| H A D | op_embedding4b_test.cpp | 39 Tensor weight_zero_points = tf.make({3}, {1, -5, 0}); in TEST() local
114 Tensor weight_zero_points = tf.make({4}, {1, 5, 7, 5}); in TEST() local
141 Tensor weight_zero_points = tf.make({2}, {1, 5}); in TEST() local
|
| /aosp_15_r20/external/executorch/exir/passes/ |
| H A D | _quant_patterns_and_replacements.py | 49 def embedding_weight_checks(weight, weight_scales, weight_zero_points): argument
856 weight_zero_points, argument
876 weight_zero_points, argument
895 weight_zero_points, argument
919 weight_zero_points, argument
939 weight_zero_points, argument
960 weight_zero_points, argument
980 weight_zero_points, argument
1005 weight_zero_points, argument
1026 weight_zero_points, argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear.cpp | 919 at::Tensor weight_zero_points, in linear_int8_with_onednn_weight()
1200 Tensor weight_zero_points, in run_pointwise()
1229 Tensor weight_zero_points, in run_pointwise_tensor()
1260 Tensor weight_zero_points, in run_pointwise_binary()
1292 Tensor weight_zero_points, in run_pointwise_binary_tensor()
|
| H A D | qconv.cpp | 1393 at::Tensor weight_zero_points, in _quantized_convolution_onednn()
1875 at::Tensor weight_zero_points, in run_pointwise()
1927 at::Tensor weight_zero_points, in run_pointwise_binary()
|
| H A D | qembeddingbag_prepack.cpp | 77 std::vector<float> weight_zero_points( in prepack() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
| H A D | qlinear_deserialize.cpp | 98 const at::Tensor weight_zero_points = in deserialize() local
|
| /aosp_15_r20/external/executorch/extension/aten_util/test/ |
| H A D | make_aten_functor_from_et_functor_test.cpp | 96 const Tensor& weight_zero_points, in quantized_embedding_byte_out()
|