diff options
author | 2018-06-20 10:27:32 -0700 | |
---|---|---|
committer | 2018-06-20 10:32:00 -0700 | |
commit | 88625ad7257ecf9d33f36f8395bf00a427a8f4e3 (patch) | |
tree | 012219e1501582e566ffb83d0f15f6dfe25c1721 /tensorflow/contrib/lite/kernels/kernel_util.h | |
parent | 856adff285f4fb271baee5603fdb623f1e32e744 (diff) |
16-bit quantized add support in TFLite interpreter
PiperOrigin-RevId: 201374318
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); |