diff options
author | 2018-06-26 23:41:11 -0700 | |
---|---|---|
committer | 2018-06-28 21:37:43 -0700 | |
commit | c30c57bd0792c50397883252ee5b2960988846d3 (patch) | |
tree | 5412a82ddd4ef7e3dde8c8389c88321e49608561 /tensorflow/contrib/lite/kernels/conv.cc | |
parent | 92cc6352abaf2442c0d29755f87f6dbcd514a684 (diff) |
add int32 support for add
PiperOrigin-RevId: 202259189
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index 93267f9a4f..2bd9e0867a 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -382,8 +382,8 @@ void EvalFloat(TfLiteContext* context, TfLiteNode* node, TfLiteTensor* filter, TfLiteTensor* bias, TfLiteTensor* im2col, TfLiteTensor* hwcn_weights, TfLiteTensor* output) { float output_activation_min, output_activation_max; - CalculateActivationRangeFloat(params->activation, &output_activation_min, - &output_activation_max); + CalculateActivationRange(params->activation, &output_activation_min, + &output_activation_max); KernelType effective_kernel_type; if (((kernel_type == kMultithreadOptimized) || (kernel_type == kCblasOptimized)) && |