Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h194 static const int idxPlanes = isColMajor ? 1 : 3; in CuboidMaxPooling() local
209 post_reduce_dims[idxPlanes] = Eigen::divup( in CuboidMaxPooling()
210 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1, in CuboidMaxPooling()
219 post_reduce_dims[idxPlanes] = Eigen::divup( in CuboidMaxPooling()
220 static_cast<DenseIndex>(in.dimension(idxPlanes)), stridePlanes); in CuboidMaxPooling()
550 static const int idxPlanes = isColMajor ? 1 : 3;
564 post_reduce_dims[idxPlanes] = Eigen::divup(
565 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1,
574 post_reduce_dims[idxPlanes] = Eigen::divup(
575 static_cast<DenseIndex>(in.dimension(idxPlanes)), stridePlanes);