Home
last modified time | relevance | path

Searched refs:patchIndex (Results 1 – 6 of 6) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorPatch.h159 Index patchIndex = index / m_outputStrides[output_stride_index];
161 Index patchOffset = index - patchIndex * m_outputStrides[output_stride_index];
165 const Index patchIdx = patchIndex / m_patchStrides[i];
166 patchIndex -= patchIdx * m_patchStrides[i];
173 const Index patchIdx = patchIndex / m_patchStrides[i];
174 patchIndex -= patchIdx * m_patchStrides[i];
180 inputIndex += (patchIndex + patchOffset);
DTensorImagePatch.h327 const Index patchIndex = index / m_fastPatchStride;
329 const Index patchOffset = (index - patchIndex * m_patchStride) / m_fastOutputDepth;
333 …const Index patch2DIndex = (NumDims == 4) ? patchIndex : (index - otherIndex * m_otherStride) / m_…
373 const Index patchIndex = indices[0] / m_fastPatchStride;
374 if (patchIndex != indices[1] / m_fastPatchStride) {
381 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
382 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth};
384 …const Index patch2DIndex = (NumDims == 4) ? patchIndex : (indices[0] - otherIndex * m_otherStride)…
DTensorVolumePatch.h356 const Index patchIndex = index / m_fastPatchStride;
360 const Index patchOffset = (index - patchIndex * m_patchStride) / m_fastOutputDepth;
364 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (index - otherIndex * m_otherStride) / m_…
420 const Index patchIndex = indices[0] / m_fastPatchStride;
421 if (patchIndex != indices[1] / m_fastPatchStride) {
428 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
429 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth};
431 …const Index patch3DIndex = (NumDims == 5) ? patchIndex : (indices[0] - otherIndex * m_otherStride)…
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h254 const Index patchIndex = index / m_fastPatchStride; in coeff() local
259 (index - patchIndex * m_patchStride) / m_fastOutputDepth; in coeff()
265 ? patchIndex in coeff()
345 const Index patchIndex = indices[0] / m_fastPatchStride; in packet() local
346 if (patchIndex != indices[1] / m_fastPatchStride) { in packet()
355 (indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth, in packet()
356 (indices[1] - patchIndex * m_patchStride) / m_fastOutputDepth}; in packet()
360 ? patchIndex in packet()
Deigen_spatial_convolutions-inl.h242 EIGEN_STRONG_INLINE Scalar operator()(Index row, Index patchIndex) const { in operator()
244 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in operator()
258 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row, Index patchIndex) const { in loadPacket() argument
260 computeBaseIndices(patchIndex, rowIndex, colIndex, otherIndex); in loadPacket()
613 Index patchIndex, Index& rowIndex, Index& colIndex, in computeBaseIndices() argument
617 otherIndex = (NumInputDims == 3) ? 0 : patchIndex / m_fastNumPatches; in computeBaseIndices()
619 ? patchIndex in computeBaseIndices()
620 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()
Deigen_cuboid_convolution.h304 EIGEN_STRONG_INLINE Scalar operator()(Index row, Index patchIndex) const { in operator()
306 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in operator()
320 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row, Index patchIndex) const { in loadPacket() argument
322 computeBaseIndices(patchIndex, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
773 Index patchIndex, Index& planeIndex, Index& rowIndex, Index& colIndex, in computeBaseIndices() argument
779 otherIndex = (NumInputDims == 4) ? 0 : patchIndex / m_fastNumPatches; in computeBaseIndices()
783 ? patchIndex in computeBaseIndices()
784 : (patchIndex - otherIndex * m_num_patches); in computeBaseIndices()