Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h263 const Index patch3DIndex = in coeff() local
269 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows; in coeff()
285 (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes; in coeff()
302 (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex)); in coeff()
358 const Index patch3DIndex = in packet() local
362 eigen_assert(patch3DIndex == in packet()
365 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows; in packet()
382 (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes; in packet()
401 (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex)); in packet()
469 const Index patch3DIndex = in coeff() local
[all …]
Deigen_cuboid_convolution.h589 const Index patch3DIndex = (NumInputDims == 4) in computeBaseIndices() local
595 colIndex = patch3DIndex / m_fastOutputPlanesRows; in computeBaseIndices()
597 (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes; in computeBaseIndices()
599 patch3DIndex - (colIndex * m_outputRows + rowIndex) * m_outputPlanes; in computeBaseIndices()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h364 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (index - otherIndex * m_otherStride) / m_…
367 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows;
377 const Index rowIndex = (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes;
387 const Index planeIndex = (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex));
431 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (indices[0] - otherIndex * m_otherStride)…
432 eigen_assert(patch3DIndex == (indices[1] - otherIndex * m_otherStride) / m_fastPatchStride);
434 const Index colIndex = patch3DIndex / m_fastOutputPlanesRows;
451 const Index rowIndex = (patch3DIndex - colIndex * m_outputPlanesRows) / m_fastOutputPlanes;
469 const Index planeIndex = (patch3DIndex - m_outputPlanes * (colIndex * m_outputRows + rowIndex));