diff options
Diffstat (limited to 'tensorflow/core/kernels/avgpooling_op.cc')
-rw-r--r-- | tensorflow/core/kernels/avgpooling_op.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/avgpooling_op.cc b/tensorflow/core/kernels/avgpooling_op.cc index af629d0de8..f918023693 100644 --- a/tensorflow/core/kernels/avgpooling_op.cc +++ b/tensorflow/core/kernels/avgpooling_op.cc @@ -153,7 +153,8 @@ class AvgPoolingOp<GPUDevice, T> : public UnaryOp<T> { if (data_format_ == FORMAT_NCHW) { DnnPoolingOp<T>::Compute( context, perftools::gputools::dnn::PoolingMode::kAverage, ksize_, - stride_, padding_, data_format_, tensor_in, output_shape); + stride_, padding_, data_format_, tensor_in, output_shape, + /*propagate_nans=*/false); } else { Tensor* output = nullptr; OP_REQUIRES_OK(context, @@ -408,7 +409,7 @@ class AvgPoolingGradOp<GPUDevice, T> : public OpKernel { DnnPoolingGradOp<T>::Compute( context, perftools::gputools::dnn::PoolingMode::kAverage, ksize_, stride_, padding_, data_format_, nullptr, nullptr, out_backprop, - output_shape); + output_shape, /*propagate_nans=*/false); } private: @@ -532,7 +533,7 @@ class AvgPoolingGradOpCustomGPUKernel : public OpKernel { DnnPoolingGradOp<T>::Compute( context, perftools::gputools::dnn::PoolingMode::kAverage, ksize_, stride_, padding_, data_format_, nullptr, nullptr, out_backprop, - output_shape); + output_shape, /*propagate_nans=*/false); } } |