Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions-inl.h270 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_rowInputStride; } in patchDepth() function
313 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeff()
337 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeffStandard()
351 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacket()
385 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPartialPacketStandard()
468 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPacketStandardFromSingleColumn()
489 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
493 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
537 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
541 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
[all …]
Deigen_cuboid_convolution.h332 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_planeInputStride; } in patchDepth() function
391 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeff()
426 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeffStandard()
444 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket()
483 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPartialPacketStandard()
574 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPacketStandardFromSingleColumnSingleRow()
598 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
601 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
665 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
668 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
[all …]
Deigen_spatial_convolutions.h116 if (standard_patches && (rhs.patchDepth() % packet_size == 0)) {
136 } else if (rhs.patchDepth() % packet_size == 0) {
207 (max_row - start_row) * rhs.patchDepth());