Home
last modified time | relevance | path

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

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorVolumePatch.h375 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (index - otherIndex * m_otherStride) / m_…
378 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows;
388 const Index rowIndex = (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes;
398 const Index planeIndex = (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex));
442 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (indices[0] - otherIndex * m_otherStride)…
443 eigen_assert(patch3DIndex == (indices[1] - otherIndex * m_otherStride) / m_fastPatchStride);
445 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows;
462 const Index rowIndex = (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes;
480 const Index planeIndex = (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex));
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_cuboid_convolution.h782 const Index patch3DIndex = (NumInputDims == 4) in computeBaseIndices() local
788 colIndex = patch3DIndex / m_fastOutputPlanesRows; in computeBaseIndices()
790 (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes; in computeBaseIndices()
792 patch3DIndex - (colIndex * m_outputRows + rowIndex) * m_outputPlanes; in computeBaseIndices()