Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h83 static const int idxRows = isColMajor ? 1 : 2; variable
96 post_reduce_dims[idxRows] = Eigen::divup(
97 static_cast<DenseIndex>(in.dimension(idxRows)) - patchRowsEff + 1,
103 post_reduce_dims[idxRows] = Eigen::divup(
104 static_cast<DenseIndex>(in.dimension(idxRows)), strideRows);
195 static const int idxRows = 2; in CuboidMaxPooling() local
212 post_reduce_dims[idxRows] = Eigen::divup( in CuboidMaxPooling()
213 static_cast<DenseIndex>(in.dimension(idxRows)) - patchRows + 1, in CuboidMaxPooling()
221 post_reduce_dims[idxRows] = Eigen::divup( in CuboidMaxPooling()
222 static_cast<DenseIndex>(in.dimension(idxRows)), strideRows); in CuboidMaxPooling()
[all …]