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()
536 CuboidAvgPooling(const Input& input, DenseIndex patchPlanes,
565 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1,
584 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes;
609 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes,
|
| D | eigen_cuboid_convolution.h | 332 EIGEN_ALWAYS_INLINE Index patchPlanes() const { return m_rowStride; } in patchPlanes() function
442 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket()
459 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
531 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketFast()
801 return std::min<Index>(1 + max_plane, patchPlanes()); in maxPlane()
828 EIGEN_ALWAYS_INLINE Index patchPlanes() const { in patchPlanes() function
|