Searched refs:pre_reduce_dims (Results 1 – 2 of 2) 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()586 Eigen::DSizes<DenseIndex, 3> pre_reduce_dims;587 pre_reduce_dims[1] = patchRows * patchCols * patchPlanes;589 pre_reduce_dims[0] = post_reduce_dims[0];[all …]
79 std::vector<int64> pre_reduce_dims; in HandleReduce() local80 pre_reduce_dims.push_back(max_reduce_dim); in HandleReduce()88 reduce->mutable_operand(1), pre_reduce_dims, reduce->to_apply()); in HandleReduce()