diff options
author | 2018-06-12 13:33:17 -0700 | |
---|---|---|
committer | 2018-06-12 13:36:24 -0700 | |
commit | 688a09dc6b70a81cae12a7e263515964311f8d86 (patch) | |
tree | 33ed3ef178fc82fb87511067df65e5789cfb0801 /tensorflow/contrib/lite/kernels/add.cc | |
parent | a22ceb68f81bb048ddd576de8ebef98d6ac1ed53 (diff) |
Standardize shifts in (more) multiplication util functions.
PiperOrigin-RevId: 200271078
Diffstat (limited to 'tensorflow/contrib/lite/kernels/add.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/add.cc | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/kernels/add.cc b/tensorflow/contrib/lite/kernels/add.cc index 7ca1e35489..443ce8924a 100644 --- a/tensorflow/contrib/lite/kernels/add.cc +++ b/tensorflow/contrib/lite/kernels/add.cc @@ -126,16 +126,19 @@ void EvalAddQuantized(TfLiteContext* context, TfLiteNode* node, int32 input1_multiplier; int input1_shift; - QuantizeMultiplierSmallerThanOne(real_input1_multiplier, &input1_multiplier, - &input1_shift); + QuantizeMultiplierSmallerThanOneExp(real_input1_multiplier, + &input1_multiplier, &input1_shift); + input1_shift *= -1; int32 input2_multiplier; int input2_shift; - QuantizeMultiplierSmallerThanOne(real_input2_multiplier, &input2_multiplier, - &input2_shift); + QuantizeMultiplierSmallerThanOneExp(real_input2_multiplier, + &input2_multiplier, &input2_shift); + input2_shift *= -1; int32 output_multiplier; int output_shift; - QuantizeMultiplierSmallerThanOne(real_output_multiplier, &output_multiplier, - &output_shift); + QuantizeMultiplierSmallerThanOneExp(real_output_multiplier, + &output_multiplier, &output_shift); + output_shift *= -1; int32 output_activation_min, output_activation_max; CalculateActivationRangeUint8(params->activation, output, |