Searched refs:newIteratorTypes (Results 1 – 4 of 4) sorted by relevance
159 SmallVector<utils::IteratorType> newIteratorTypes; in splitReduction() local163 newIteratorTypes.push_back(utils::IteratorType::parallel); in splitReduction()164 newIteratorTypes.push_back(iteratorType); in splitReduction()167 newIteratorTypes.push_back(utils::IteratorType::parallel); in splitReduction()173 ValueRange({identityTensor}), newMaps, newIteratorTypes); in splitReduction()
471 SmallVector<utils::IteratorType> newIteratorTypes = in mergeReductions() 474 newIteratorTypes[dim] = utils::IteratorType::parallel; in mergeReductions() 479 tiledInits, newMaps, newIteratorTypes); in mergeReductions() 441 SmallVector<utils::IteratorType> newIteratorTypes = tileToPartialReduction() local
422 SmallVector<utils::IteratorType> newIteratorTypes; in dropUnitDims() 432 newIteratorTypes.push_back(attr); in dropUnitDims() 523 newIndexingMaps, newIteratorTypes); in dropUnitDims() 424 SmallVector<utils::IteratorType> newIteratorTypes; dropUnitDims() local
351 SmallVector<Attribute> newIteratorTypes; in castAwayContractionLeadingOneDim() local 363 newIteratorTypes.push_back(it.value()); in castAwayContractionLeadingOneDim() 458 rewriter.getArrayAttr(newIteratorTypes), contractOp.getKind()); in castAwayContractionLeadingOneDim()