Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp193 SmallVector<StringRef> newIteratorTypes; in splitReduction() local
196 newIteratorTypes.push_back(getParallelIteratorTypeName()); in splitReduction()
197 newIteratorTypes.push_back(it.value().cast<StringAttr>().getValue()); in splitReduction()
203 ValueRange({identityTensor}), newMaps, newIteratorTypes); in splitReduction()
H A DDropUnitDims.cpp207 SmallVector<Attribute, 4> newIteratorTypes; in matchAndRewrite() local
210 newIteratorTypes.push_back(attr.value()); in matchAndRewrite()
215 genericOp.iterator_typesAttr(ArrayAttr::get(context, newIteratorTypes)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp308 SmallVector<Attribute> newIteratorTypes; in matchAndRewrite() local
320 newIteratorTypes.push_back(it.value()); in matchAndRewrite()
395 rewriter.getArrayAttr(newIteratorTypes), contractOp.getKind()); in matchAndRewrite()