diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-12 13:33:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-12 13:36:24 -0700 |
commit | 688a09dc6b70a81cae12a7e263515964311f8d86 (patch) | |
tree | 33ed3ef178fc82fb87511067df65e5789cfb0801 /tensorflow/contrib/lite/kernels/conv.cc | |
parent | a22ceb68f81bb048ddd576de8ebef98d6ac1ed53 (diff) |
Standardize shifts in (more) multiplication util functions.
PiperOrigin-RevId: 200271078
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index 747c8a62c0..14b399ef96 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -257,8 +257,9 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { TF_LITE_ENSURE_STATUS(GetQuantizedConvolutionMultipler( context, input, filter, bias, output, &real_multiplier)); TF_LITE_ENSURE(context, real_multiplier < 1.0); - QuantizeMultiplierSmallerThanOne(real_multiplier, &data->output_multiplier, - &data->output_shift); + QuantizeMultiplierSmallerThanOneExp( + real_multiplier, &data->output_multiplier, &data->output_shift); + data->output_shift *= -1; CalculateActivationRangeUint8(params->activation, output, &data->output_activation_min, &data->output_activation_max); |