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.h264 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_rowInputStride; } in patchDepth() function
307 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeff()
331 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeffStandard()
345 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacket()
359 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
363 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
396 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPacketStandard()
412 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketFast()
415 eigen_assert((patchDepth() % packetSize) == 0); in loadPacketFast()
430 const Index depth = patchId - patchOffset * patchDepth(); in loadPacketFast()
[all …]
Deigen_cuboid_convolution.h330 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_planeInputStride; } in patchDepth() function
389 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeff()
424 const Index depth = patchId - patchOffset * patchDepth(); in loadCoeffStandard()
442 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket()
459 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
462 if ((patchDepth() % packetSize) == 0) { in loadPacketStandard()
509 const Index depth = patchId - patchOffsets[0] * patchDepth(); in loadPacketStandard()
531 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketFast()
534 eigen_assert((patchDepth() % packetSize) == 0); in loadPacketFast()
556 const Index depth = patchId - patchOffset * patchDepth(); in loadPacketFast()
[all …]
Deigen_spatial_convolutions.h64 if (standard_patches && (rhs.patchDepth() % packet_size == 0)) {
125 (max_row - start_row) * rhs.patchDepth());