/external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
D | fully_connected_test.cc | 110 float output_max, TfLiteFusedActivation activation, uint8_t* output_data) { in TestFullyConnectedQuantized() argument 128 output_min, output_max), in TestFullyConnectedQuantized() 273 const float output_max = 128.0f; in TF_LITE_MICRO_TEST() local 321 F2Q(24, output_min, output_max), in TF_LITE_MICRO_TEST() 322 F2Q(25, output_min, output_max), in TF_LITE_MICRO_TEST() 323 F2Q(26, output_min, output_max), in TF_LITE_MICRO_TEST() 324 F2Q(58, output_min, output_max), in TF_LITE_MICRO_TEST() 325 F2Q(59, output_min, output_max), in TF_LITE_MICRO_TEST() 326 F2Q(60, output_min, output_max), in TF_LITE_MICRO_TEST() 329 output_min, output_max, // Output quantization range. in TF_LITE_MICRO_TEST() [all …]
|
D | depthwise_conv_test.cc | 117 float output_max, TfLiteFusedActivation activation, uint8_t* output_data) { in TestDepthwiseConvQuantized() argument 135 output_min, output_max), in TestDepthwiseConvQuantized() 235 const float output_max = 128.0f; in TF_LITE_MICRO_TEST() local 289 F2Q(71, output_min, output_max), in TF_LITE_MICRO_TEST() 290 F2Q(-34, output_min, output_max), in TF_LITE_MICRO_TEST() 291 F2Q(99, output_min, output_max), in TF_LITE_MICRO_TEST() 292 F2Q(-20, output_min, output_max), in TF_LITE_MICRO_TEST() 293 F2Q(91, output_min, output_max), in TF_LITE_MICRO_TEST() 294 F2Q(-26, output_min, output_max), in TF_LITE_MICRO_TEST() 295 F2Q(127, output_min, output_max), in TF_LITE_MICRO_TEST() [all …]
|
D | softmax_test.cc | 94 float output_min, float output_max, in TestSoftmaxQuantized() argument 107 output_min, output_max), in TestSoftmaxQuantized() 194 const float output_max = (255.0f / 256.0f); in TF_LITE_MICRO_TEST() local 209 F2Q(0.011656231, output_min, output_max), in TF_LITE_MICRO_TEST() 210 F2Q(0.031684921, output_min, output_max), in TF_LITE_MICRO_TEST() 211 F2Q(0.086128544, output_min, output_max), in TF_LITE_MICRO_TEST() 212 F2Q(0.234121657, output_min, output_max), in TF_LITE_MICRO_TEST() 213 F2Q(0.636408647, output_min, output_max), in TF_LITE_MICRO_TEST() 216 output_min, output_max, // Output quantized range. in TF_LITE_MICRO_TEST()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_add_op.cc | 48 float scalar_input_max, float output_min, float output_max, in ScalarAddition() argument 51 scalar_input, scalar_input_min, scalar_input_max, output_min, output_max); in ScalarAddition() 54 full_input[i], full_input_min, full_input_max, output_min, output_max); in ScalarAddition() 66 float output_min, float output_max, qint32* output) { in ScalarAddition() argument 68 scalar_input, scalar_input_min, scalar_input_max, output_min, output_max); in ScalarAddition() 75 FloatToQuantizedUnclamped<qint32>(input_0_float, output_min, output_max); in ScalarAddition() 77 FloatToQuantizedUnclamped<qint32>(input_1_float, output_min, output_max); in ScalarAddition() 123 float output_min, float output_max, qint32* output) { in ScalarAddition() argument 125 scalar_input, scalar_input_min, scalar_input_max, output_min, output_max); in ScalarAddition() 132 FloatToQuantizedUnclamped<qint32>(input_0_float, output_min, output_max); in ScalarAddition() [all …]
|
D | quantized_batch_norm_op.cc | 38 Tensor* output, float* output_min, float* output_max) { in ReferenceBatchNorm() argument 50 *output_max = std::numeric_limits<float>::lowest(); in ReferenceBatchNorm() 79 *output_max = std::max(output_value, *output_max); in ReferenceBatchNorm() 82 FloatToQuantized<T2>(output_value, *output_min, *output_max); in ReferenceBatchNorm() 101 Tensor* output, float* output_min, float* output_max) { in FixedPointBatchNorm() argument 116 *output_max = (1 << 20); in FixedPointBatchNorm() 139 FloatToQuantized<T2>(scale_value, *output_min, *output_max); in FixedPointBatchNorm() 141 FloatToQuantized<T2>(offset_value, *output_min, *output_max); in FixedPointBatchNorm() 145 FloatToQuantized<T2>(1.0f, *output_min, *output_max); in FixedPointBatchNorm() 151 input_max, *output_min, *output_max); in FixedPointBatchNorm() [all …]
|
D | quantized_concat_op.cc | 35 float output_min, float output_max) in RequantizeCopier() 37 output_max(output_max), in RequantizeCopier() 43 if (input_min == output_min && input_max == output_max) { in Copy() 54 FloatToQuantizedStruct<T> f2q(output_min, output_max); in Copy() 65 float output_max; member 82 float* output_min, float* output_max) { in CalculateInputAndOutputRange() argument 101 *output_max = largest_value; in CalculateInputAndOutputRange() 104 *output_max = overall_max; in CalculateInputAndOutputRange() 195 float output_max = std::numeric_limits<float>::lowest(); in Compute() local 199 &output_max); in Compute() [all …]
|
D | quantized_activation_ops.cc | 54 Tensor* output_max = nullptr; in Compute() local 55 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute() 56 output_max->flat<float>()(0) = max_input; in Compute() 92 Tensor* output_max = nullptr; in Compute() local 93 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute() 94 output_max->flat<float>()(0) = max_input; in Compute()
|
D | quantized_pooling_ops.cc | 97 Tensor* output_max = nullptr; in Compute() local 98 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute() 99 output_max->flat<float>()(0) = max_input; in Compute() 121 Tensor* output_max = nullptr; in Compute() local 122 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute() 123 output_max->flat<float>()(0) = max_input; in Compute()
|
D | quantization_utils_test.cc | 35 float input_max, float output_min, float output_max, in TestRequantizeMany() argument 44 output_min, output_max)); in TestRequantizeMany() 55 input_max, output_min, output_max, in TestRequantizeMany() 59 *eigen_device, i_tensor, input_min, input_max, output_min, output_max, in TestRequantizeMany() 72 << ", output_max=" << output_max << ", value_index=" << value_index; in TestRequantizeMany() 77 float output_min, float output_max, in TestRequantizeMany8To32Bit() argument 86 output_min, output_max)); in TestRequantizeMany8To32Bit() 96 input_max, output_min, output_max, in TestRequantizeMany8To32Bit() 108 << ", output_max=" << output_max << ", value_index=" << value_index; in TestRequantizeMany8To32Bit() 233 const float output_max = ranges[range_index][3]; in TestRequantizeManyInNewRangeEigenVsNonEigen() local [all …]
|
D | mkl_requantize_ops_test.cc | 164 const float output_max = GetOutput(1)->flat<float>()(0); in TEST_F() local 166 EXPECT_NEAR(14.8217, output_max, 0.002); in TEST_F() 225 const float output_max = GetOutput(1)->flat<float>()(0); in TEST_F() local 227 EXPECT_NEAR(6.0, output_max, 0.002); // Values are aligned with clip_value. in TEST_F() 294 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 296 EXPECT_NEAR(range_op_output_max, output_max, 0.002); in TEST_F()
|
D | quantization_utils.cc | 23 float* output_min, float* output_max) { in GetOutputMinAndMaxForQuantizedAdd() argument 35 *output_max = in GetOutputMinAndMaxForQuantizedAdd() 39 *output_min = -(*output_max); in GetOutputMinAndMaxForQuantizedAdd()
|
D | quantized_activation_ops_test.cc | 63 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 65 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F() 94 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 96 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F()
|
D | quantized_reshape_op.cc | 40 Tensor* output_max = nullptr; in Compute() local 41 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute() 42 output_max->flat<float>()(0) = input_max_float; in Compute()
|
D | quantized_pooling_ops_test.cc | 77 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 79 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F() 122 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 124 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F()
|
D | requantization_range_op.cc | 53 Tensor* output_max = nullptr; in Compute() local 54 OP_REQUIRES_OK(ctx, ctx->allocate_output(1, TensorShape({}), &output_max)); in Compute() 69 output_max->flat<float>().setConstant(used_max_float); in Compute()
|
D | quantized_bias_add_op_test.cc | 84 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 86 QuantizedTensorToFloat<qint32>(output_quantized, output_min, output_max); in TEST_F() 166 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 168 QuantizedTensorToFloat<qint32>(output_quantized, output_min, output_max); in TEST_F()
|
D | quantize_op_test.cc | 293 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 295 EXPECT_NEAR(255.0f, output_max, 1e-5f); in TEST_F() 315 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 317 EXPECT_LT(0.0f, output_max); in TEST_F() 337 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 339 EXPECT_NEAR(0.3f, output_max, 1e-5f); in TEST_F() 359 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 361 EXPECT_NEAR(0.0f, output_max, 1e-5f); in TEST_F()
|
D | requantize.cc | 51 Tensor* output_max = nullptr; in Compute() local 52 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute() 93 output_max->flat<float>().setConstant(requested_output_max_float); in Compute()
|
D | quantized_concat_op_test.cc | 112 const float output_max = GetOutput(2)->flat<float>()(0); in TestSmall8Bit() local 114 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TestSmall8Bit() 178 const float output_max = GetOutput(2)->flat<float>()(0); in TestSmall32Bit() local 180 QuantizedTensorToFloat<qint32>(output_quantized, output_min, output_max); in TestSmall32Bit() 240 const float output_max = GetOutput(2)->flat<float>()(0); in TestSecondDim8Bit() local 242 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TestSecondDim8Bit()
|
D | mkl_quantized_pooling_ops_test.cc | 126 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 128 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F() 194 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() local 196 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TEST_F()
|
D | quantize_down_and_shrink_range.cc | 49 Tensor* output_max = nullptr; in Compute() local 50 OP_REQUIRES_OK(ctx, ctx->allocate_output(2, TensorShape({}), &output_max)); in Compute() 97 output_max->flat<float>().setConstant(actual_max_float); in Compute()
|
D | mkl_requantization_range_per_channel_op.cc | 103 Tensor* output_max = nullptr; in Compute() local 105 OP_REQUIRES_OK(ctx, ctx->allocate_output(kOutputMaxIndex, {}, &output_max)); in Compute() 107 output_max->flat<float>()(0) = out_min_max; in Compute()
|
D | quantized_bias_add_op.cc | 88 Tensor* output_max = nullptr; in Compute() local 89 OP_REQUIRES_OK(context, context->allocate_output(2, {}, &output_max)); in Compute() 90 output_max->flat<float>()(0) = total_max; in Compute()
|
D | mkl_quantized_concat_op_test.cc | 151 const float output_max = GetOutput(2)->flat<float>()(0); in TestSmall8Bit() local 153 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TestSmall8Bit() 225 const float output_max = GetOutput(2)->flat<float>()(0); in TestSecondDim8Bit() local 227 QuantizedTensorToFloat<quint8>(output_quantized, output_min, output_max); in TestSecondDim8Bit()
|
D | meta_support.h | 81 float output_max, quint8* output); 101 float bias_max, float output_min, float output_max,
|