Searched refs:reductionDims (Results 1 – 4 of 4) sorted by relevance
49 auto reductionDims = llvm::to_vector<4>(llvm::map_range( in matchAndRewrite() local51 llvm::SmallDenseSet<int64_t> reductionDimsSet(reductionDims.begin(), in matchAndRewrite()52 reductionDims.end()); in matchAndRewrite()53 int64_t reductionSize = reductionDims.size(); in matchAndRewrite()76 indices.append(reductionDims.begin(), reductionDims.end()); in matchAndRewrite()78 indices.append(reductionDims.begin(), reductionDims.end()); in matchAndRewrite()129 SmallVector<int64_t, 4> reductionDims, reductionShapes; in matchAndRewrite() local134 reductionDims.push_back(i); in matchAndRewrite()166 counter = reductionDims.size(); in matchAndRewrite()
1018 SmallVector<int64_t> reductionDims; in getCollapsableIterationSpaceDims() local1021 reductionDims.push_back(iteratorType.index()); in getCollapsableIterationSpaceDims()1062 for (const auto &startDim : llvm::enumerate(reductionDims)) { in getCollapsableIterationSpaceDims()1069 reductionDims.size()) in getCollapsableIterationSpaceDims()1075 if (reductionDims[foldedDim.index() + startDim.index()] != in getCollapsableIterationSpaceDims()
339 SmallVector<int64_t> reductionDims; in build() local342 reductionDims.push_back(en.index()); in build()344 builder.getI64ArrayAttr(reductionDims)); in build()
371 ArrayRef<int64_t> reductionDims, unsigned sourceRank) {373 for (auto idx : reductionDims)