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()
553 static const int idxPlanes = isColMajor ? 1 : 3;
567 post_reduce_dims[idxPlanes] = Eigen::divup(
568 static_cast<DenseIndex>(in.dimension(idxPlanes)) - patchPlanes + 1,
577 post_reduce_dims[idxPlanes] = Eigen::divup(
578 static_cast<DenseIndex>(in.dimension(idxPlanes)), stridePlanes);