Home
last modified time | relevance | path

Searched refs:m_row_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),
100 DenseIndex row_strides() const { return m_row_strides; }
134 const DenseIndex m_row_strides;
194 m_row_strides = op.row_strides();
221 …padding_top() + op.padding_bottom() - m_patch_rows_eff + 1.f) / static_cast<float>(m_row_strides));
229 …ws = numext::ceil((m_input_rows_eff - m_patch_rows_eff + 1.f) / static_cast<float>(m_row_strides));
232 …m_rowPaddingTop = numext::maxi<Index>(0, ((m_outputRows - 1) * m_row_strides + m_patch_rows_eff - …
236 m_outputRows = numext::ceil(m_input_rows_eff / static_cast<float>(m_row_strides));
239 … m_rowPaddingTop = ((m_outputRows - 1) * m_row_strides + m_patch_rows_eff - m_input_rows_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),
100 DenseIndex row_strides() const { return m_row_strides; }
144 const DenseIndex m_row_strides;
210 m_row_strides = op.row_strides();
231 …padding_top() + op.padding_bottom() - m_patch_rows_eff + 1.f) / static_cast<float>(m_row_strides));
241 …ws = numext::ceil((m_input_rows_eff - m_patch_rows_eff + 1.f) / static_cast<float>(m_row_strides));
249 m_outputRows = numext::ceil(m_input_rows_eff / static_cast<float>(m_row_strides));
252 const Index dy = m_outputRows * m_row_strides + m_patch_rows_eff - 1 - m_input_rows_eff;
379 … const Index inputRow = rowIndex * m_row_strides + rowOffset * m_in_row_strides - m_rowPaddingTop;
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h76 m_row_strides = op.row_strides(); in CustomTensorEvaluator()
108 m_row_strides); in CustomTensorEvaluator()
124 m_row_strides); in CustomTensorEvaluator()
133 m_outputRows = Eigen::divup(m_input_rows_eff, m_row_strides); in CustomTensorEvaluator()
139 0, (m_outputRows - 1) * m_row_strides + m_patch_rows_eff - in CustomTensorEvaluator()
288 const Index inputRow = rowIndex * m_row_strides + in coeff()
389 rowIndex * m_row_strides + rowOffsets[0] - m_rowPaddingTop, in packet()
390 rowIndex * m_row_strides + rowOffsets[1] - m_rowPaddingTop}; in packet()
452 Index userRowStride() const { return m_row_strides; } in userRowStride()
495 const Index inputRow = rowIndex * m_row_strides + in coeff()
[all …]
Deigen_spatial_convolutions-inl.h128 m_row_strides = tensor.impl().userRowStride(); in TensorContractionInputMapper()
186 m_row_strides = base_mapper.m_row_strides; in TensorContractionInputMapper()
461 rowIndex = rowIndex * m_row_strides - m_rowPaddingTop; in computeBaseIndices()
495 Index m_row_strides; // User specified row stride variable
726 return m_base_mapper.m_row_strides; in rowStride()
Deigen_cuboid_convolution.h140 m_row_strides = tensor.impl().userRowStride(); in TensorContractionInputMapper()
238 m_row_strides = base_mapper.m_row_strides; in TensorContractionInputMapper()
602 rowIndex = rowIndex * m_row_strides - m_rowPaddingTop; in computeBaseIndices()
640 Index m_row_strides; // User specified row stride variable