Searched refs:patchRows (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_pooling.h | 64 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() 421 SpatialAvgPooling(const Input& input, DenseIndex patchRows, 435 patchRows + (patchRows - 1) * (in_strideRows - 1); 491 .extract_image_patches(patchRows, patchCols, strideRows, strideCols, [all …]
|
D | eigen_spatial_convolutions-inl.h | 266 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_colStride; } in patchRows() function 345 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacket() 359 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard() 412 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketFast() 625 return std::min<Index>(1 + max_row, patchRows()); in maxRow() 657 EIGEN_ALWAYS_INLINE Index patchRows() const { in patchRows() function
|
D | eigen_cuboid_convolution.h | 334 EIGEN_ALWAYS_INLINE Index patchRows() const { return m_patch_rows; } in patchRows() function 442 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket() 459 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard() 531 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketFast() 792 return std::min<Index>(1 + max_row, patchRows()); in maxRow() 834 EIGEN_ALWAYS_INLINE Index patchRows() const { in patchRows() function
|