Searched refs:pre_reduce_dims (Results 1 – 1 of 1) sorted by relevance
228 Eigen::DSizes<DenseIndex, 3> pre_reduce_dims; in CuboidMaxPooling() local229 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 …]