Home
last modified time | relevance | path

Searched defs:weight_scales (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/executorch/kernels/quantized/test/
H A Dop_embedding_test.cpp54 Tensor weight_scales = tf.full({3}, scale); in test_dtype() local
113 Tensor weight_scales = tf.full({3}, scale); in TEST() local
182 Tensor weight_scales = tf.make({3}, {0.5, 1.0, 1.5}); in TEST() local
239 Tensor weight_scales = tf.make({4}, {0.5, 1.0, 1.5, 3.3}); in TEST() local
269 Tensor weight_scales = tf.make({2}, {0.5, 1.0}); in TEST() local
299 Tensor weight_scales = tf.make({3, 2}, {0.5, 1.0, 1.5, 2.5, 3.5, 3.5}); in TEST() local
328 Tensor weight_scales = tf.make({3, 2}, {0.5, 1.0, 1.5, 2.5, 3.5, 3.5}); in TEST() local
357 Tensor weight_scales = tf.make({3, 2}, {0.5, 1.0, 1.5, 2.5, 3.5, 3.5}); in TEST() local
H A Dop_embedding2b_test.cpp38 Tensor weight_scales = tf.make({3}, {0.5, 1.0, 1.5}); in TEST() local
116 Tensor weight_scales = tf.make({4}, {0.5, 1.0, 1.5, 3.3}); in TEST() local
144 Tensor weight_scales = tf.make({2}, {0.5, 1.0}); in TEST() local
172 Tensor weight_scales = tf.make({2, 3}, {1.0, 1.0, 1.0, 1.0, 1.0, 1.0}); in TEST() local
H A Dop_embedding4b_test.cpp38 Tensor weight_scales = tf.make({3}, {0.5, 1.0, 1.5}); in TEST() local
113 Tensor weight_scales = tf.make({4}, {0.5, 1.0, 1.5, 3.3}); in TEST() local
140 Tensor weight_scales = tf.make({2}, {0.5, 1.0}); in TEST() local
H A Dop_mixed_linear_test.cpp48 Tensor weight_scales = tf.make( in test_dtype() local
101 Tensor weight_scales = tf.make( in test_dtype_partials() local
H A Dop_mixed_mm_test.cpp47 Tensor weight_scales = tf.make( in test_dtype() local
/aosp_15_r20/external/executorch/kernels/quantized/cpu/
H A Dop_embedding.cpp29 const Tensor& weight_scales, in check_embedding_byte_args()
131 const Tensor& weight_scales, in embedding_byte_per_channel()
220 const Tensor& weight_scales, in quantized_embedding_byte_out()
255 const Tensor& weight_scales, in quantized_embedding_byte_out()
279 const Tensor& weight_scales, in quantized_embedding_byte_dtype_out()
318 const Tensor& weight_scales, in quantized_embedding_byte_dtype_out()
H A Dembeddingxb.cpp67 const Tensor& weight_scales, in check_embedding_xbit_args()
172 const Tensor& weight_scales, in embedding_xbit_per_channel()
262 const Tensor& weight_scales, in quantized_embedding_xbit_out()
301 const Tensor& weight_scales, in quantized_embedding_xbit_out()
327 const Tensor& weight_scales, in quantized_embedding_xbit_dtype_out()
370 const Tensor& weight_scales, in quantized_embedding_xbit_dtype_out()
H A Dop_embedding4b.cpp39 const Tensor& weight_scales, in quantized_embedding_4bit_out()
59 const Tensor& weight_scales, in quantized_embedding_4bit_out()
81 const Tensor& weight_scales, in quantized_embedding_4bit_dtype_out()
103 const Tensor& weight_scales, in quantized_embedding_4bit_dtype_out()
H A Dop_embedding2b.cpp39 const Tensor& weight_scales, in quantized_embedding_2bit_out()
59 const Tensor& weight_scales, in quantized_embedding_2bit_out()
79 const Tensor& weight_scales, in quantized_embedding_2bit_dtype_out()
101 const Tensor& weight_scales, in quantized_embedding_2bit_dtype_out()
H A Dop_mixed_mm.cpp21 const Tensor& weight_scales, in check_quantized_mixed_mm_args()
57 const Tensor& weight_scales, in quantized_mixed_mm_out()
94 const Tensor& weight_scales, in quantized_mixed_mm_out()
H A Dop_mixed_linear.cpp21 const Tensor& weight_scales, in check_quantized_mixed_linear_args()
66 const Tensor& weight_scales, in quantized_mixed_linear_out()
119 const Tensor& weight_scales, in quantized_mixed_linear_out()
/aosp_15_r20/external/executorch/exir/passes/
H A D_quant_patterns_and_replacements.py49 def embedding_weight_checks(weight, weight_scales, weight_zero_points): argument
855 weight_scales, argument
875 weight_scales, argument
894 weight_scales, argument
918 weight_scales, argument
938 weight_scales, argument
959 weight_scales, argument
979 weight_scales, argument
1004 weight_scales, argument
1025 weight_scales, argument
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp918 at::Tensor weight_scales, in linear_int8_with_onednn_weight()
1199 Tensor weight_scales, in run_pointwise()
1228 Tensor weight_scales, in run_pointwise_tensor()
1259 Tensor weight_scales, in run_pointwise_binary()
1291 Tensor weight_scales, in run_pointwise_binary_tensor()
H A DQnnpackUtils.h448 const at::Tensor& weight_scales, in generate_requantization_scales()
499 at:: Tensor weight_scales = variable
H A Dqconv_prepack.cpp495 at::Tensor weight_scales, // Weight zero points must be 0 for onednn in _qconv_prepack_onednn()
810 at::Tensor weight_scales, // Weight zero points must be 0s for onednn in run_conv()
H A Dqconv.cpp1392 at::Tensor weight_scales, in _quantized_convolution_onednn()
1874 at::Tensor weight_scales, in run_pointwise()
1926 at::Tensor weight_scales, in run_pointwise_binary()
H A Dqembeddingbag_prepack.cpp74 std::vector<float> weight_scales( in prepack() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h410 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
420 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h414 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
424 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantization_utils.cc267 std::vector<float> weight_scales(channel_dim_size); in AdjustWeightsForBiasScale() local
671 const float* weight_scales, in SymmetricPerChannelBiasQuantize()
H A Dquantization_utils_test.cc855 const std::vector<float> weight_scales = {0.5, 1.0}; in TEST_F() local
/aosp_15_r20/external/executorch/backends/xnnpack/test/runtime/
H A Dtest_runtime_utils.cpp70 const Tensor weight_scales = tf.full({3, 5}, 4.0); in TEST() local
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/utils/
H A Dutils.cpp147 const Tensor& weight_scales, in GenerateRequantizationScale()
/aosp_15_r20/external/executorch/extension/aten_util/test/
H A Dmake_aten_functor_from_et_functor_test.cpp95 const Tensor& weight_scales, in quantized_embedding_byte_out()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.cc555 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()
575 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()

12