Home
last modified time | relevance | path

Searched refs:m_col_strides (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h74 m_row_strides(row_strides), m_col_strides(col_strides),
88 m_row_strides(row_strides), m_col_strides(col_strides),
102 DenseIndex col_strides() const { return m_col_strides; }
135 const DenseIndex m_col_strides;
195 m_col_strides = op.col_strides();
222 …padding_left() + op.padding_right() - m_patch_cols_eff + 1.f) / static_cast<float>(m_col_strides));
230 …ls = numext::ceil((m_input_cols_eff - m_patch_cols_eff + 1.f) / static_cast<float>(m_col_strides));
233 …m_colPaddingLeft = numext::maxi<Index>(0, ((m_outputCols - 1) * m_col_strides + m_patch_cols_eff -…
237 m_outputCols = numext::ceil(m_input_cols_eff / static_cast<float>(m_col_strides));
240 … m_colPaddingLeft = ((m_outputCols - 1) * m_col_strides + m_patch_cols_eff - m_input_cols_eff) / 2;
[all …]
DTensorVolumePatch.h69 m_plane_strides(plane_strides), m_row_strides(row_strides), m_col_strides(col_strides),
84 m_plane_strides(plane_strides), m_row_strides(row_strides), m_col_strides(col_strides),
102 DenseIndex col_strides() const { return m_col_strides; }
145 const DenseIndex m_col_strides;
211 m_col_strides = op.col_strides();
232 …padding_left() + op.padding_right() - m_patch_cols_eff + 1.f) / static_cast<float>(m_col_strides));
242 …ls = numext::ceil((m_input_cols_eff - m_patch_cols_eff + 1.f) / static_cast<float>(m_col_strides));
250 m_outputCols = numext::ceil(m_input_cols_eff / static_cast<float>(m_col_strides));
253 const Index dx = m_outputCols * m_col_strides + m_patch_cols_eff - 1 - m_input_cols_eff;
369 … const Index inputCol = colIndex * m_col_strides + colOffset * m_in_col_strides - m_colPaddingLeft;
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h77 m_col_strides = op.col_strides(); in CustomTensorEvaluator()
113 m_col_strides); in CustomTensorEvaluator()
126 m_col_strides); in CustomTensorEvaluator()
134 m_outputCols = Eigen::divup(m_input_cols_eff, m_col_strides); in CustomTensorEvaluator()
142 0, (m_outputCols - 1) * m_col_strides + m_patch_cols_eff - in CustomTensorEvaluator()
271 const Index inputCol = colIndex * m_col_strides + in coeff()
371 colIndex * m_col_strides + colOffsets[0] - m_colPaddingLeft, in packet()
372 colIndex * m_col_strides + colOffsets[1] - m_colPaddingLeft}; in packet()
453 Index userColStride() const { return m_col_strides; } in userColStride()
481 const Index inputCol = colIndex * m_col_strides + in coeff()
[all …]
Deigen_spatial_convolutions-inl.h129 m_col_strides = tensor.impl().userColStride(); in TensorContractionInputMapper()
187 m_col_strides = base_mapper.m_col_strides; in TensorContractionInputMapper()
460 colIndex = colIndex * m_col_strides - m_colPaddingLeft; in computeBaseIndices()
496 Index m_col_strides; // User specified col stride variable
730 return m_base_mapper.m_col_strides; in colStride()
Deigen_cuboid_convolution.h141 m_col_strides = tensor.impl().userColStride(); in TensorContractionInputMapper()
239 m_col_strides = base_mapper.m_col_strides; in TensorContractionInputMapper()
601 colIndex = colIndex * m_col_strides - m_colPaddingLeft; in computeBaseIndices()
641 Index m_col_strides; // User specified col stride variable