diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-01 13:34:55 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-01 13:41:55 -0700 |
commit | 4963bcf071dea91bb1fca38d1c281d9d76a7dfa3 (patch) | |
tree | 76ea7a131ff9f8f3516ce2d30f2629e1857eda4c /tensorflow/contrib/lite/kernels/conv.cc | |
parent | 9354ab24b9760382b6e0fcd5f3276191fd27f233 (diff) |
Generalize the Conv() op to support input product scale > output scale, i.e.
output_multiplier > 1.
#20451
#19607
PiperOrigin-RevId: 206983654
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index 6f174763df..04c0263b78 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -256,10 +256,10 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { double real_multiplier = 0.0; TF_LITE_ENSURE_STATUS(GetQuantizedConvolutionMultipler( context, input, filter, bias, output, &real_multiplier)); - TF_LITE_ENSURE(context, real_multiplier < 1.0); - QuantizeMultiplierSmallerThanOneExp( - real_multiplier, &data->output_multiplier, &data->output_shift); - data->output_shift *= -1; + + int exponent; + QuantizeMultiplier(real_multiplier, &data->output_multiplier, &exponent); + data->output_shift = -exponent; CalculateActivationRangeUint8(params->activation, output, &data->output_activation_min, &data->output_activation_max); |