Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h65 DenseIndex patchCols, DenseIndex strideRows,
80 patchCols + (patchCols - 1) * (in_strideCols - 1);
132 patchRows, patchCols, strideRows, strideCols, in_strideRows,
181 DenseIndex patchRows, DenseIndex patchCols, in CuboidMaxPooling() argument
216 static_cast<DenseIndex>(in.dimension(idxCols)) - patchCols + 1, in CuboidMaxPooling()
229 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes; in CuboidMaxPooling()
250 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes, in CuboidMaxPooling()
422 DenseIndex patchCols, DenseIndex strideRows,
437 patchCols + (patchCols - 1) * (in_strideCols - 1);
491 .extract_image_patches(patchRows, patchCols, strideRows, strideCols,
[all …]
Deigen_cuboid_convolution.h336 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols() function
442 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacket()
459 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketStandard()
531 patchDepth() * patchPlanes() * patchRows() * patchCols()); in loadPacketFast()
784 return std::min<Index>(1 + max_col, patchCols()); in maxCol()
838 EIGEN_ALWAYS_INLINE Index patchCols() const { in patchCols() function
Deigen_spatial_convolutions-inl.h268 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols() function
616 return std::min<Index>(1 + max_col, patchCols()); in maxCol()
661 EIGEN_ALWAYS_INLINE Index patchCols() const { in patchCols() function