Searched refs:dims_mask (Results 1 – 1 of 1) sorted by relevance
1758 llvm::SmallVector<bool, 4> dims_mask(rank, false); in GetReduceResultType() local1759 for (int64_t dim : dimensions.getValues<int64_t>()) dims_mask[dim] = true; in GetReduceResultType()1763 if (!dims_mask[i]) shape.push_back(ranked_ty.getDimSize(i)); in GetReduceResultType()