Home
last modified time | relevance | path

Searched refs:patchOffsets (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorPatch.h205 Index patchOffsets[2] = {indices[0] - patchIndices[0] * m_outputStrides[output_stride_index],
217 const Index offsetIdx[2] = {patchOffsets[0] / m_outputStrides[i],
218 patchOffsets[1] / m_outputStrides[i]};
219 patchOffsets[0] -= offsetIdx[0] * m_outputStrides[i];
220 patchOffsets[1] -= offsetIdx[1] * m_outputStrides[i];
233 const Index offsetIdx[2] = {patchOffsets[0] / m_outputStrides[i+1],
234 patchOffsets[1] / m_outputStrides[i+1]};
235 patchOffsets[0] -= offsetIdx[0] * m_outputStrides[i+1];
236 patchOffsets[1] -= offsetIdx[1] * m_outputStrides[i+1];
242 inputIndices[0] += (patchIndices[0] + patchOffsets[0]);
[all …]
H A DTensorVolumePatch.h439 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
447 patchOffsets[0] / m_fastColStride,
448 patchOffsets[1] / m_fastColStride};
464 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride,
465 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride};
482 patchOffsets[0] - colOffsets[0] * m_colStride - rowOffsets[0] * m_rowStride,
483 patchOffsets[1] - colOffsets[1] * m_colStride - rowOffsets[1] * m_rowStride};
H A DTensorImagePatch.h466 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
473 …const Index colOffsets[2] = {patchOffsets[0] / m_fastColStride, patchOffsets[1] / m_fastColStride};
484 …const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0]*m_colStride, patchOffsets[1] - colOff…
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_cuboid_convolution.h466 Index patchId, const Index span[], const Index patchOffsets[], in loadPartialPacketStandard() argument
471 patchOffsets[0] - colOffset * m_colStride - rowOffset * m_rowStride, in loadPartialPacketStandard()
472 patchOffsets[1] - colOffset * m_colStride - rowOffset * m_rowStride}; in loadPartialPacketStandard()
483 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPartialPacketStandard()
515 Index otherIndex, const Index patchOffsets[], const Index colOffsets[], in loadPacketStandardFromSingleColumnTwoRows() argument
538 {patchOffsets[0], patchOffsetSplit}, in loadPacketStandardFromSingleColumnTwoRows()
539 {patchOffsetSplit + 1, patchOffsets[1]}}; in loadPacketStandardFromSingleColumnTwoRows()
556 Index otherIndex, const Index patchOffsets[], const Index colOffsets[], in loadPacketStandardFromSingleColumnSingleRow() argument
562 patchOffsets[0] - colOffsets[0] * m_colStride - in loadPacketStandardFromSingleColumnSingleRow()
564 patchOffsets[1] - colOffsets[1] * m_colStride - in loadPacketStandardFromSingleColumnSingleRow()
[all …]
H A Deigen_spatial_convolutions-inl.h372 const Index span[], const Index patchOffsets[], Index colOffset) const { in loadPartialPacketStandard() argument
374 const Index rowOffsets[2] = {patchOffsets[0] - colOffset * m_colStride, in loadPartialPacketStandard()
375 patchOffsets[1] - colOffset * m_colStride}; in loadPartialPacketStandard()
386 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPartialPacketStandard()
417 const Index patchOffsets[], const Index colOffsets[]) const { in loadPacketStandardFromTwoColumns() argument
435 {patchOffsets[0], patchOffsetSplit}, in loadPacketStandardFromTwoColumns()
436 {patchOffsetSplit + 1, patchOffsets[1]}}; in loadPacketStandardFromTwoColumns()
453 const Index patchOffsets[], const Index colOffsets[], in loadPacketStandardFromSingleColumn() argument
456 const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0] * m_colStride, in loadPacketStandardFromSingleColumn()
457 patchOffsets[1] - colOffsets[1] * m_colStride}; in loadPacketStandardFromSingleColumn()
[all …]