diff options
Diffstat (limited to 'tensorflow/core/kernels/pooling_ops_common.cc')
-rw-r--r-- | tensorflow/core/kernels/pooling_ops_common.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/pooling_ops_common.cc b/tensorflow/core/kernels/pooling_ops_common.cc index ddc9c9823b..3fe16c66b8 100644 --- a/tensorflow/core/kernels/pooling_ops_common.cc +++ b/tensorflow/core/kernels/pooling_ops_common.cc @@ -64,6 +64,8 @@ PoolParameters::PoolParameters(OpKernelContext* context, OP_REQUIRES_OK( context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, padding, &out_width, &pad_cols)); + pad_depth = 0; + out_depth = depth; } else { // Our current version of depthwise max pooling does not support // any padding, and expects the depth_window to equal the |