Home
last modified time | relevance | path

Searched refs:FloatToQuantized (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantization_utils_test.cc42 expected_values.push_back(FloatToQuantized<quint8>( in TestRequantizeMany()
84 expected_values.push_back(FloatToQuantized<qint32>( in TestRequantizeMany8To32Bit()
135 values_quantized.push_back(FloatToQuantized<qint32>(v, r[0], r[1])); in TestRequantizeManyInNewRange32To8Bit()
166 qint32 low = FloatToQuantized<qint32>(out_min, in_min, in_max); in TestRequantizeManyInNewRange32To8Bit()
167 qint32 high = FloatToQuantized<qint32>(out_max, in_min, in_max); in TestRequantizeManyInNewRange32To8Bit()
193 values_quantized.push_back(FloatToQuantized<quint8>(v, r[0], r[1])); in TestRequantizeManyInNewRange8To32Bit()
241 FloatToQuantized<InputType>(v, input_min, input_max)); in TestRequantizeManyInNewRangeEigenVsNonEigen()
293 FloatToQuantized<InputType>(v, input_min, input_max)); in TimeRequantizeManyInNewRange()
366 int32_t expected = FloatToQuantized<T>(input_array(i), f_min, f_max); in TestFloatToQuantizedInPlaceUsingEigen()
434 EXPECT_EQ(quint8(0), FloatToQuantized<quint8>(0.0f, 0.0f, 1.0f)); in TestFloatToQuantized()
[all …]
H A Dquantized_activation_ops.cc53 const T min_as_quantized = FloatToQuantized<T>(0.0f, min_input, max_input); in Compute()
99 const T min_as_quantized = FloatToQuantized<T>(0.0f, min_input, max_input); in Compute()
100 const T max_as_quantized = FloatToQuantized<T>(6.0f, min_input, max_input); in Compute()
H A Dquantized_batch_norm_op.cc82 FloatToQuantized<T2>(output_value, *output_min, *output_max); in ReferenceBatchNorm()
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()
H A Dquantization_utils.h73 T FloatToQuantized(float input, float range_min, float range_max) { in FloatToQuantized() function
231 return FloatToQuantized<T2>(input_float, min_new, max_new); in RequantizeInNewRange()
242 output[index] = FloatToQuantized<T2>(input_float, min_output, max_output); in RequantizeManyInNewRange()
756 flat_result(i) = FloatToQuantized<T>(flat_input(i), min, max); in FloatTensorToQuantizedInPlace()
828 FloatToQuantized<T3>(0.0f, *output_min, *output_max); in QuantizedAddUsingEigen()
874 FloatToQuantized<T3>(0.0f, *output_min, *output_max); in QuantizedAdd()
H A Dquantized_resize_bilinear_op_test.cc46 FloatToQuantized<T>(static_cast<float>(i) / ratio, min, max); in BuildTensor()
95 return FloatToQuantized<T>(out, min, max); in ComputeLerpReference()
253 FloatToQuantized<qint32>(static_cast<float>(i), MIN, MAX); in TestResizeBilinearOneDim()
H A Dquantized_resize_bilinear_op.cc114 return FloatToQuantized<T>(out, min, max); in ComputeLerpReference()