diff --git a/tensorflow/core/kernels/avgpooling_op.cc b/tensorflow/core/kernels/avgpooling_op.cc index 58004d1789f..654446bf99b 100644 --- a/tensorflow/core/kernels/avgpooling_op.cc +++ b/tensorflow/core/kernels/avgpooling_op.cc @@ -77,6 +77,11 @@ class AvgPoolingOp : public UnaryOp { OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1, errors::Unimplemented( "Pooling is not yet supported on the batch dimension.")); + + for (int i = 0; i < ksize_.size(); ++i) { + OP_REQUIRES(context, ksize_[i] != 0, + errors::InvalidArgument("ksize cannot be zero")); + } } void Compute(OpKernelContext* context) override {