Skip to content
Permalink
Browse files Browse the repository at this point in the history
Fix divide by zero error in fractional_pool_common.cc.
PiperOrigin-RevId: 371126221
Change-Id: Iea4b2f363aaeb116ab460e3bc592c687484af344
  • Loading branch information
pak-laura authored and tensorflower-gardener committed Apr 29, 2021
1 parent acc8ee6 commit 548b5ea
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tensorflow/core/kernels/fractional_avg_pool_op.cc
Expand Up @@ -80,6 +80,10 @@ class FractionalAvgPoolOp : public OpKernel {
std::vector<int> output_size(tensor_in_and_out_dims);
for (int i = 0; i < tensor_in_and_out_dims; ++i) {
input_size[i] = tensor_in.dim_size(i);
OP_REQUIRES(
context, pooling_ratio_[i] <= input_size[i],
errors::InvalidArgument(
"Pooling ratio cannot be bigger than input tensor dim size."));
}
// Output size.
for (int i = 0; i < tensor_in_and_out_dims; ++i) {
Expand Down

0 comments on commit 548b5ea

Please sign in to comment.