diff options
Diffstat (limited to 'tensorflow/core/kernels/quantization_utils.h')
-rw-r--r-- | tensorflow/core/kernels/quantization_utils.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/quantization_utils.h b/tensorflow/core/kernels/quantization_utils.h index cb4fcbd788..c5dc2e7194 100644 --- a/tensorflow/core/kernels/quantization_utils.h +++ b/tensorflow/core/kernels/quantization_utils.h @@ -823,9 +823,9 @@ void QuantizedAddUsingEigen(const Eigen::ThreadPoolDevice& device, const int64 input_element_count = input.NumElements(); const int64 smaller_input_element_count = smaller_input.NumElements(); - QuantizedToFloatStruct<T1> smaller_input_q2f(smaller_input_min, + QuantizedToFloatStruct<T1> input_q2f(input_min, input_max); + QuantizedToFloatStruct<T2> smaller_input_q2f(smaller_input_min, smaller_input_max); - QuantizedToFloatStruct<T2> input_q2f(input_min, input_max); FloatToQuantizedStruct<T3> f2q(*output_min, *output_max); auto smaller_input_float = |