Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h84 static const int idxCols = isColMajor ? 2 : 1; variable
99 post_reduce_dims[idxCols] = Eigen::divup(
100 static_cast<DenseIndex>(in.dimension(idxCols)) - patchColsEff + 1,
105 post_reduce_dims[idxCols] = Eigen::divup(
106 static_cast<DenseIndex>(in.dimension(idxCols)), strideCols);
196 static const int idxCols = isColMajor ? 3 : 1; in CuboidMaxPooling() local
215 post_reduce_dims[idxCols] = Eigen::divup( in CuboidMaxPooling()
216 static_cast<DenseIndex>(in.dimension(idxCols)) - patchCols + 1, in CuboidMaxPooling()
223 post_reduce_dims[idxCols] = Eigen::divup( in CuboidMaxPooling()
224 static_cast<DenseIndex>(in.dimension(idxCols)), strideCols); in CuboidMaxPooling()
[all …]