diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-26 23:41:11 -0700 |
---|---|---|
committer | Gunhan Gulsoy <gunan@google.com> | 2018-06-28 21:37:43 -0700 |
commit | c30c57bd0792c50397883252ee5b2960988846d3 (patch) | |
tree | 5412a82ddd4ef7e3dde8c8389c88321e49608561 /tensorflow/contrib/lite/kernels/pooling.cc | |
parent | 92cc6352abaf2442c0d29755f87f6dbcd514a684 (diff) |
add int32 support for add
PiperOrigin-RevId: 202259189
Diffstat (limited to 'tensorflow/contrib/lite/kernels/pooling.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/pooling.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/kernels/pooling.cc b/tensorflow/contrib/lite/kernels/pooling.cc index 58d74c97a7..7240fe04cc 100644 --- a/tensorflow/contrib/lite/kernels/pooling.cc +++ b/tensorflow/contrib/lite/kernels/pooling.cc @@ -124,8 +124,8 @@ void AverageEvalFloat(TfLiteContext* context, TfLiteNode* node, TfLitePoolParams* params, OpData* data, const TfLiteTensor* input, TfLiteTensor* output) { float activation_min, activation_max; - CalculateActivationRangeFloat(params->activation, &activation_min, - &activation_max); + CalculateActivationRange(params->activation, &activation_min, + &activation_max); #define TF_LITE_AVERAGE_POOL(type) \ type::AveragePool(GetTensorData<float>(input), GetTensorShape(input), \ params->stride_width, params->stride_height, \ @@ -169,8 +169,8 @@ void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node, TfLitePoolParams* params, OpData* data, const TfLiteTensor* input, TfLiteTensor* output) { float activation_min, activation_max; - CalculateActivationRangeFloat(params->activation, &activation_min, - &activation_max); + CalculateActivationRange(params->activation, &activation_min, + &activation_max); #define TF_LITE_MAX_POOL(type) \ type::MaxPool(GetTensorData<float>(input), GetTensorShape(input), \ params->stride_width, params->stride_height, \ @@ -214,8 +214,8 @@ void L2EvalFloat(TfLiteContext* context, TfLiteNode* node, TfLitePoolParams* params, OpData* data, const TfLiteTensor* input, TfLiteTensor* output) { float activation_min, activation_max; - CalculateActivationRangeFloat(params->activation, &activation_min, - &activation_max); + CalculateActivationRange(params->activation, &activation_min, + &activation_max); #define TF_LITE_L2_POOL(type) \ type::L2Pool(GetTensorData<float>(input), GetTensorShape(input), \ params->stride_width, params->stride_height, \ |