Home
last modified time | relevance | path

Searched refs:output_scales (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc206 float output_scales[] = {1, output_scale}; in TestDepthwiseConvQuantizedPerChannel() local
208 TfLiteAffineQuantization output_quant = {FloatArrayFromFloats(output_scales), in TestDepthwiseConvQuantizedPerChannel()
746 float output_scales[2] = {1, output_scale}; in TF_LITE_MICRO_TEST() local
748 tflite::testing::FloatArrayFromFloats(output_scales), in TF_LITE_MICRO_TEST()
984 float output_scales[] = {1, output_scale}; in TF_LITE_MICRO_TEST() local
986 tflite::testing::FloatArrayFromFloats(output_scales), in TF_LITE_MICRO_TEST()
Dconv_test_common.cc222 float output_scales[] = {1, output_scale}; in TestConvQuantizedPerChannel() local
224 TfLiteAffineQuantization output_quant = {FloatArrayFromFloats(output_scales), in TestConvQuantizedPerChannel()
Dconv_test.cc496 float output_scales[2] = {1, output_scale}; in TF_LITE_MICRO_TEST() local
498 tflite::testing::FloatArrayFromFloats(output_scales), in TF_LITE_MICRO_TEST()
713 float output_scales[] = {1, output_scale}; in TF_LITE_MICRO_TEST() local
715 tflite::testing::FloatArrayFromFloats(output_scales), in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils_test.cc198 std::vector<float> output_scales(3); in TEST_F() local
207 &output_scales, &output_data, in TEST_F()
216 EXPECT_THAT(output_scales, ElementsAreArray(expected_output_scales)); in TEST_F()
230 std::vector<float> output_scales(3); in TEST_F() local
249 &output_scales, &output_data, in TEST_F()
258 EXPECT_THAT(output_scales, ElementsAreArray(expected_output_scales)); in TEST_F()
Dquantization_utils.h65 std::vector<float>* output_scales,
Dquantization_utils.cc294 std::vector<float>* output_scales, in SymmetricPerChannelQuantization() argument
319 output_scales->at(channel_idx) = half_range / half_scale; in SymmetricPerChannelQuantization()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc161 float output_scales[] = {1, output_scale}; in TestDepthwiseConvQuantizedPerChannel() local
164 FloatArrayFromFloats(output_scales), in TestDepthwiseConvQuantizedPerChannel()
Dconv_slicing_test.cc246 float output_scales[] = {1, output_scale}; in TestConvQuantizedPerChannel() local
249 FloatArrayFromFloats(output_scales), in TestConvQuantizedPerChannel()