33 static bool isShapePreserving(ForOp forOp, int64_t arg) {  in isShapePreserving()  argument
34   auto yieldOp = cast<YieldOp>(forOp.getBody()->getTerminator());  in isShapePreserving()
39     if (value == forOp.getRegionIterArgs()[arg])  in isShapePreserving()
50             .template Case<ForOp>([&](ForOp forOp) {  in isShapePreserving()  argument
51               return isShapePreserving(forOp, opResult.getResultNumber())  in isShapePreserving()
52                          ? forOp.getIterOperands()[opResult.getResultNumber()]  in isShapePreserving()
92     auto forOp = dyn_cast<ForOp>(blockArg.getParentBlock()->getParentOp());  in matchAndRewrite()  local
93     if (!forOp)  in matchAndRewrite()
95     if (!isShapePreserving(forOp, blockArg.getArgNumber() - 1))  in matchAndRewrite()
98     Value initArg = forOp.getOpOperandForRegionIterArg(blockArg).get();  in matchAndRewrite()
134     auto forOp = dimOp.getSource().template getDefiningOp<scf::ForOp>();  in matchAndRewrite()  local
135     if (!forOp)  in matchAndRewrite()
139     if (!isShapePreserving(forOp, resultNumber))  in matchAndRewrite()
142       dimOp.getSourceMutable().assign(forOp.getIterOperands()[resultNumber]);  in matchAndRewrite()
158       if (scf::ForOp forOp = scf::getForInductionVarOwner(iv)) {  in matchAndRewrite()  local
159         lb = forOp.getLowerBound();  in matchAndRewrite()
160         ub = forOp.getUpperBound();  in matchAndRewrite()
161         step = forOp.getStep();  in matchAndRewrite()