Home
last modified time | relevance | path

Searched refs:patch2DIndex (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorImagePatch.h418 …const Index patch2DIndex = (NumDims == 4) ? patchIndex : (index - otherIndex * m_otherStride) / m_…
421 const Index colIndex = patch2DIndex / m_fastOutputRows;
431 const Index rowIndex = patch2DIndex - colIndex * m_outputRows;
469 …const Index patch2DIndex = (NumDims == 4) ? patchIndex : (indices[0] - otherIndex * m_otherStride)…
470 eigen_assert(patch2DIndex == (indices[1] - otherIndex * m_otherStride) / m_fastPatchStride);
472 const Index colIndex = patch2DIndex / m_fastOutputRows;
483 const Index rowIndex = patch2DIndex - colIndex * m_outputRows;
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_spatial_convolutions-inl.h619 const Index patch2DIndex = (NumInputDims == 3) in computeBaseIndices() local
623 colIndex = patch2DIndex / m_fastOutputRows; in computeBaseIndices()
624 rowIndex = patch2DIndex - colIndex * m_outputRows; in computeBaseIndices()