diff options
author | Yangzihao Wang <yangzihao@google.com> | 2017-12-12 11:15:24 -0800 |
---|---|---|
committer | Yangzihao Wang <yangzihao@google.com> | 2017-12-12 11:15:24 -0800 |
commit | 3122477c8660f4e66e9cf4bf24e4fdfd6d56378c (patch) | |
tree | 898cd8cbe292cfabe779b4cbfead5da57958bbbe /unsupported/Eigen | |
parent | 393b7c495912dcf45491ef8f131c2599f1b5b147 (diff) |
Update the padding computation for PADDING_SAME to be consistent with TensorFlow.
Diffstat (limited to 'unsupported/Eigen')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h b/unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h index 3c6a2e091..91d4ead28 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h @@ -265,6 +265,10 @@ struct TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>, Device> // Calculate the padding m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + m_patch_rows_eff - m_input_rows_eff) / 2; m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + m_patch_cols_eff - m_input_cols_eff) / 2; + // The padding size calculation for PADDING_SAME has been updated to + // be consistent with how TensorFlow extracts its paddings. + m_rowPaddingTop = numext::maxi<Index>(0, m_rowPaddingTop); + m_colPaddingLeft = numext::maxi<Index>(0, m_colPaddingLeft); break; default: eigen_assert(false && "unexpected padding"); |