Searched refs:loopLikeOp (Results 1 – 5 of 5) sorted by relevance
59 auto loopLikeOp = cast<LoopLikeOpInterface>(op); in verifyLoopLikeOpInterface() local62 if (loopLikeOp.getInits().size() != loopLikeOp.getRegionIterArgs().size()) in verifyLoopLikeOpInterface()64 << loopLikeOp.getInits().size() in verifyLoopLikeOpInterface()65 << " != " << loopLikeOp.getRegionIterArgs().size(); in verifyLoopLikeOpInterface()66 if (!loopLikeOp.getYieldedValues().empty() && in verifyLoopLikeOpInterface()67 loopLikeOp.getRegionIterArgs().size() != in verifyLoopLikeOpInterface()68 loopLikeOp.getYieldedValues().size()) in verifyLoopLikeOpInterface()71 << loopLikeOp.getRegionIterArgs().size() in verifyLoopLikeOpInterface()72 << " != " << loopLikeOp.getYieldedValues().size(); in verifyLoopLikeOpInterface()73 if (loopLikeOp.getLoopResults() && loopLikeOp.getLoopResults()->size() != in verifyLoopLikeOpInterface()[all …]
29 void checkUnidimensional(LoopLikeOpInterface loopLikeOp) { in checkUnidimensional() argument31 loopLikeOp.getSingleLowerBound(); in checkUnidimensional()34 loopLikeOp.getSingleUpperBound(); in checkUnidimensional()36 std::optional<OpFoldResult> maybeSingleStep = loopLikeOp.getSingleStep(); in checkUnidimensional()39 loopLikeOp.getSingleInductionVar(); in checkUnidimensional()43 loopLikeOp.getLoopLowerBounds(); in checkUnidimensional()47 loopLikeOp.getLoopUpperBounds(); in checkUnidimensional()51 loopLikeOp.getLoopSteps(); in checkUnidimensional()55 loopLikeOp.getLoopInductionVars(); in checkUnidimensional()60 void checkMultidimensional(LoopLikeOpInterface loopLikeOp) { in checkMultidimensional() argument[all …]
25 transform::TransformRewriter &rewriter, LoopLikeOpInterface loopLikeOp, in applyToOne() argument28 hoistLoopInvariantSubsets(rewriter, loopLikeOp); in applyToOne()
70 ::mlir::LoopLikeOpInterface loopLikeOp,
846 LoopLikeOpInterface loopLikeOp, RewriterBase &rewriter, in tileAndFuseProducerOfSlice() 849 loopLikeOp.getOperation()) in tileAndFuseProducerOfSlice() 442 yieldTiledValuesAndReplaceLoop(LoopLikeOpInterface loopLikeOp,RewriterBase & rewriter,ValueRange newInitOperands,YieldTiledValuesFn yieldTiledValuesFn) yieldTiledValuesAndReplaceLoop() argument