mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-08-13 12:19:12 +08:00
Fix compilation in TensorImagePatch
This commit is contained in:
parent
38874b1651
commit
9ab8ac5c8b
@ -142,15 +142,15 @@ struct TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>, Device>
|
|||||||
|
|
||||||
switch (op.padding_type()) {
|
switch (op.padding_type()) {
|
||||||
case PADDING_VALID:
|
case PADDING_VALID:
|
||||||
m_outputRows = ceil((m_inputRows - op.patch_rows() + 1.f) / static_cast<float>(m_row_strides));
|
m_outputRows = numext::ceil((m_inputRows - op.patch_rows() + 1.f) / static_cast<float>(m_row_strides));
|
||||||
m_outputCols = ceil((m_inputCols - op.patch_cols() + 1.f) / static_cast<float>(m_col_strides));
|
m_outputCols = numext::ceil((m_inputCols - op.patch_cols() + 1.f) / static_cast<float>(m_col_strides));
|
||||||
// Calculate the padding
|
// Calculate the padding
|
||||||
m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + op.patch_rows() - m_inputRows) / 2;
|
m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + op.patch_rows() - m_inputRows) / 2;
|
||||||
m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + op.patch_cols() - m_inputCols) / 2;
|
m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + op.patch_cols() - m_inputCols) / 2;
|
||||||
break;
|
break;
|
||||||
case PADDING_SAME:
|
case PADDING_SAME:
|
||||||
m_outputRows = ceil(m_inputRows / static_cast<float>(m_row_strides));
|
m_outputRows = numext::ceil(m_inputRows / static_cast<float>(m_row_strides));
|
||||||
m_outputCols = ceil(m_inputCols / static_cast<float>(m_col_strides));
|
m_outputCols = numext::ceil(m_inputCols / static_cast<float>(m_col_strides));
|
||||||
// Calculate the padding
|
// Calculate the padding
|
||||||
m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + op.patch_rows() - m_inputRows) / 2;
|
m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + op.patch_rows() - m_inputRows) / 2;
|
||||||
m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + op.patch_cols() - m_inputCols) / 2;
|
m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + op.patch_cols() - m_inputCols) / 2;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user