Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h64 SpatialMaxPooling(const Input& input, DenseIndex patchRows,
78 patchRows + (patchRows - 1) * (in_strideRows - 1);
132 patchRows, patchCols, strideRows, strideCols, in_strideRows,
181 DenseIndex patchRows, DenseIndex patchCols, in CuboidMaxPooling() argument
213 static_cast<DenseIndex>(in.dimension(idxRows)) - patchRows + 1, in CuboidMaxPooling()
229 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes; in CuboidMaxPooling()
250 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes, in CuboidMaxPooling()
424 SpatialAvgPooling(const Input& input, DenseIndex patchRows,
438 patchRows + (patchRows - 1) * (in_strideRows - 1);
494 .extract_image_patches(patchRows, patchCols, strideRows, strideCols,
[all …]
Deigen_spatial_convolutions-inl.h272 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_colStride; } in patchRows() function
351 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacket()
489 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
537 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
576 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketFast()
793 return std::min<Index>(1 + max_row, patchRows()); in maxRow()
825 EIGEN_ALWAYS_INLINE Index patchRows() const { in patchRows() function
890 if (last_row + (patchRows() - 1) >= m_base_mapper.m_inputRows) return true; in hasPadding()
Deigen_cuboid_convolution.h336 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_patch_rows; } in patchRows() 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()
986 return std::min<Index>(1 + max_row, patchRows()); in maxRow()
1028 EIGEN_ALWAYS_INLINE Index patchRows() const { in patchRows() function