Lines Matching defs:forOp

38 static bool isShapePreserving(ForOp forOp, int64_t arg) {
39   assert(arg < static_cast<int64_t>(forOp.getNumResults()) &&
41   Value value = forOp.getYieldedValues()[arg];
43     if (value == forOp.getRegionIterArgs()[arg])
53                 .template Case<ForOp>([&](ForOp forOp) {
54                   return isShapePreserving(forOp, opResult.getResultNumber())
55                              ? forOp.getInitArgs()[opResult.getResultNumber()]
95     auto forOp = dyn_cast<ForOp>(blockArg.getParentBlock()->getParentOp());
96     if (!forOp)
98     if (!isShapePreserving(forOp, blockArg.getArgNumber() - 1))
101     Value initArg = forOp.getTiedLoopInit(blockArg)->get();
137     auto forOp = dimOp.getSource().template getDefiningOp<scf::ForOp>();
138     if (!forOp)
142     if (!isShapePreserving(forOp, resultNumber))
145       dimOp.getSourceMutable().assign(forOp.getInitArgs()[resultNumber]);