Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorPatch.h194 Index patchOffsets[2] = {indices[0] - patchIndices[0] * m_outputStrides[output_stride_index],
205 const Index offsetIdx[2] = {patchOffsets[0] / m_outputStrides[i],
206 patchOffsets[1] / m_outputStrides[i]};
207 patchOffsets[0] -= offsetIdx[0] * m_outputStrides[i];
208 patchOffsets[1] -= offsetIdx[1] * m_outputStrides[i];
220 const Index offsetIdx[2] = {patchOffsets[0] / m_outputStrides[i+1],
221 patchOffsets[1] / m_outputStrides[i+1]};
222 patchOffsets[0] -= offsetIdx[0] * m_outputStrides[i+1];
223 patchOffsets[1] -= offsetIdx[1] * m_outputStrides[i+1];
229 inputIndices[0] += (patchIndices[0] + patchOffsets[0]);
[all …]
DTensorVolumePatch.h428 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
436 patchOffsets[0] / m_fastColStride,
437 patchOffsets[1] / m_fastColStride};
453 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride,
454 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride};
471 patchOffsets[0] - colOffsets[0] * m_colStride - rowOffsets[0] * m_rowStride,
472 patchOffsets[1] - colOffsets[1] * m_colStride - rowOffsets[1] * m_rowStride};
DTensorImagePatch.h381 const Index patchOffsets[2] = {(indices[0] - patchIndex * m_patchStride) / m_fastOutputDepth,
388 …const Index colOffsets[2] = {patchOffsets[0] / m_fastColStride, patchOffsets[1] / m_fastColStride};
399 …const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0]*m_colStride, patchOffsets[1] - colOff…
/external/tensorflow/tensorflow/core/kernels/
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 …]
Deigen_spatial_convolutions-inl.h371 const Index span[], const Index patchOffsets[], Index colOffset) const { in loadPartialPacketStandard() argument
373 const Index rowOffsets[2] = {patchOffsets[0] - colOffset * m_colStride, in loadPartialPacketStandard()
374 patchOffsets[1] - colOffset * m_colStride}; in loadPartialPacketStandard()
385 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPartialPacketStandard()
416 const Index patchOffsets[], const Index colOffsets[]) const { in loadPacketStandardFromTwoColumns() argument
434 {patchOffsets[0], patchOffsetSplit}, in loadPacketStandardFromTwoColumns()
435 {patchOffsetSplit + 1, patchOffsets[1]}}; in loadPacketStandardFromTwoColumns()
452 const Index patchOffsets[], const Index colOffsets[], in loadPacketStandardFromSingleColumn() argument
455 const Index rowOffsets[2] = {patchOffsets[0] - colOffsets[0] * m_colStride, in loadPacketStandardFromSingleColumn()
456 patchOffsets[1] - colOffsets[1] * m_colStride}; in loadPacketStandardFromSingleColumn()
[all …]
Deigen_volume_patch.h354 const Index patchOffsets[2] = { in packet() local
366 const Index colOffsets[2] = {patchOffsets[0] / m_fastColStride, in packet()
367 patchOffsets[1] / m_fastColStride}; in packet()
384 (patchOffsets[0] - colOffsets[0] * m_colStride) / m_fastRowStride, in packet()
385 (patchOffsets[1] - colOffsets[1] * m_colStride) / m_fastRowStride}; in packet()
403 patchOffsets[0] - colOffsets[0] * m_colStride - in packet()
405 patchOffsets[1] - colOffsets[1] * m_colStride - in packet()