diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/kernel_util.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/kernel_util.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/kernel_util.h b/tensorflow/contrib/lite/kernels/kernel_util.h index 82cded36f2..20058a5f69 100644 --- a/tensorflow/contrib/lite/kernels/kernel_util.h +++ b/tensorflow/contrib/lite/kernels/kernel_util.h @@ -88,6 +88,11 @@ TfLiteStatus GetQuantizedConvolutionMultipler(TfLiteContext* context, // Calculates the useful range of an activation layer given its activation // tensor. +TfLiteStatus CalculateActivationRangeQuantized(TfLiteContext* context, + TfLiteFusedActivation activation, + TfLiteTensor* output, + int32_t* act_min, + int32_t* act_max); void CalculateActivationRangeUint8(TfLiteFusedActivation activation, TfLiteTensor* output, int32_t* act_min, int32_t* act_max); |