| /external/tensorflow/tensorflow/lite/micro/ |
| D | testing_helpers_test.cc | 45 float weight_scales[] = {0.5, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
59 pre_quantized, quantized, dims, input_scale, weight_scales, scales, in TF_LITE_MICRO_TEST()
65 TF_LITE_MICRO_EXPECT_EQ(scales[i + 1], input_scale * weight_scales[i]); in TF_LITE_MICRO_TEST()
|
| D | test_helpers.h | 211 float input_scale, float* weight_scales, float* scales, int* zero_points,
|
| D | test_helpers.cc | 1020 float input_scale, float* weight_scales, float* scales, int* zero_points, in CreatePerChannelQuantizedBiasTensor() argument
1030 scales_array[i] = input_scale * weight_scales[i]; in CreatePerChannelQuantizedBiasTensor()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 257 std::vector<float> weight_scales(channel_dim_size); in AdjustWeightsForBiasScale() local
259 quant_params, &weight_scales, error_reporter)); in AdjustWeightsForBiasScale()
266 0.5 * input_scale * weight_scales[i] * kScale) { in AdjustWeightsForBiasScale()
279 if (bias_half_range / kScale >= 0.5 * input_scale * weight_scales[0]) { in AdjustWeightsForBiasScale()
658 const float* weight_scales, in SymmetricPerChannelBiasQuantize() argument
664 scales[i] = input_scale * weight_scales[i]; in SymmetricPerChannelBiasQuantize()
689 const float* weight_scales, int number_of_dimension,
694 const float* weight_scales, int number_of_dimension,
|
| D | quantize_model.cc | 153 std::vector<float> weight_scales = weight_tensor->quantization->scale; in QuantizeBias() local
170 if (weight_scales.size() != channel_dim_size) { in QuantizeBias()
173 weight_scales.size()); in QuantizeBias()
179 weight_scales.data(), channel_dim_size, error_reporter); in QuantizeBias()
183 weight_scales.data(), channel_dim_size, error_reporter); in QuantizeBias()
186 if (weight_scales.size() != 1) { in QuantizeBias()
190 weight_scales.size()); in QuantizeBias()
196 input_tensor->quantization->scale[0] * weight_scales[0], in QuantizeBias()
201 input_tensor->quantization->scale[0] * weight_scales[0], in QuantizeBias()
|
| D | quantization_utils.h | 130 const float* weight_scales,
|
| D | quantization_utils_test.cc | 786 const std::vector<float> weight_scales = {0.5, 1.0}; in TEST_F() local
805 weight_scales.data(), 2, &error_reporter_), in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 417 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
427 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 441 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
451 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1577 const DeviceMemory<float>& weight_scales,
1589 const DeviceMemory<float>& weight_scales,
|
| D | stream.cc | 765 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized() argument
769 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(weight_scales), in ThenMatMulQuantized()
774 CheckError(dnn->DoMatMulQuantized(this, input_data, weights, weight_scales, in ThenMatMulQuantized()
785 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized() argument
789 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(weight_scales), in ThenMatMulQuantized()
794 CheckError(dnn->DoMatMulQuantized(this, input_data, weights, weight_scales, in ThenMatMulQuantized()
|
| D | stream.h | 523 const DeviceMemory<float> &weight_scales,
530 const DeviceMemory<float> &weight_scales,
|