Searched refs:patchCols (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling.h | 51 DenseIndex patchCols, DenseIndex strideRows,
66 patchCols + (patchCols - 1) * (in_strideCols - 1);
106 patchRows, patchCols, strideRows, strideCols, in_strideRows,
143 DenseIndex patchRows, DenseIndex patchCols, in CuboidMaxPooling() argument
178 static_cast<DenseIndex>(in.dimension(idxCols)) - patchCols + 1, in CuboidMaxPooling()
191 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes; in CuboidMaxPooling()
206 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes, in CuboidMaxPooling()
376 DenseIndex patchCols, DenseIndex strideRows,
391 patchCols + (patchCols - 1) * (in_strideCols - 1);
433 .extract_image_patches(patchRows, patchCols, strideRows, strideCols,
[all …]
|
| D | eigen_cuboid_convolution.h | 338 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols() 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()
978 return std::min<Index>(1 + max_col, patchCols()); in maxCol()
1032 EIGEN_ALWAYS_INLINE Index patchCols() const { in patchCols() function
|
| D | eigen_spatial_convolutions-inl.h | 275 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols() function
785 return std::min<Index>(1 + max_col, patchCols()); in maxCol()
830 EIGEN_ALWAYS_INLINE Index patchCols() const { in patchCols() function
896 if (last_col + (patchCols() - 1) >= m_base_mapper.m_inputCols) return true; in hasPadding()
|