Searched refs:patchPlanes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling.h | 180 CuboidMaxPooling(const Input& input, DenseIndex patchPlanes, in CuboidMaxPooling() argument
210 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1, in CuboidMaxPooling()
229 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes; in CuboidMaxPooling()
250 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes, in CuboidMaxPooling()
539 CuboidAvgPooling(const Input& input, DenseIndex patchPlanes,
568 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1,
587 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes;
612 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes,
|
| D | eigen_cuboid_convolution.h | 334 EIGEN_ALWAYS_INLINE Index patchPlanes() const { return m_rowStride; } in patchPlanes() function
444 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket()
598 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
665 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
724 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketFast()
995 return std::min<Index>(1 + max_plane, patchPlanes()); in maxPlane()
1022 EIGEN_ALWAYS_INLINE Index patchPlanes() const { in patchPlanes() function
|