Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h228 Eigen::DSizes<DenseIndex, 3> pre_reduce_dims; in CuboidMaxPooling() local
229 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes; in CuboidMaxPooling()
231 pre_reduce_dims[0] = post_reduce_dims[0]; in CuboidMaxPooling()
232 pre_reduce_dims[2] = post_reduce_dims[1] * post_reduce_dims[2] * in CuboidMaxPooling()
235 pre_reduce_dims[0] = post_reduce_dims[0] * post_reduce_dims[1] * in CuboidMaxPooling()
237 pre_reduce_dims[2] = post_reduce_dims[4]; in CuboidMaxPooling()
253 .reshape(pre_reduce_dims) in CuboidMaxPooling()
583 Eigen::DSizes<DenseIndex, 3> pre_reduce_dims;
584 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes;
586 pre_reduce_dims[0] = post_reduce_dims[0];
[all …]