Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h93 post_reduce_dims; variable
94 post_reduce_dims[0] = in.dimension(0);
96 post_reduce_dims[idxRows] = Eigen::divup(
99 post_reduce_dims[idxCols] = Eigen::divup(
103 post_reduce_dims[idxRows] = Eigen::divup(
105 post_reduce_dims[idxCols] = Eigen::divup(
108 post_reduce_dims[3] = in.dimension(3);
137 .reshape(post_reduce_dims);
206 post_reduce_dims; in CuboidMaxPooling() local
207 post_reduce_dims[0] = in.dimension(0); in CuboidMaxPooling()
[all …]