diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-23 16:33:27 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-23 16:35:37 -0700 |
| commit | 3a17101171d3e51fcba2189d09416c5106bfe4ac (patch) | |
| tree | 029b6268d5c6a27306ed73ca41cfa54d61ff6baa /tensorflow/contrib/lite/kernels/conv.cc | |
| parent | fa5c52e31f30e8cb88a5452e3b4aefc786fb8852 (diff) | |
Make depthwiseconv handler handle filter ranges beyond 255
PiperOrigin-RevId: 197810361
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index 0b35a220e7..ee42e5cdc8 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -254,6 +254,7 @@ 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); QuantizeMultiplierSmallerThanOne(real_multiplier, &data->output_multiplier, &data->output_shift); CalculateActivationRangeUint8(params->activation, output, |