Lines Matching defs:whileOp

100     auto whileOp = cast<scf::WhileOp>(conditionOp->getParentOp());
110             whileOp.getAfterArguments()[it.index()], options);
833     auto whileOp = cast<scf::WhileOp>(op);
843     OpResult opResult = whileOp->getResult(idx);
855     auto whileOp = cast<scf::WhileOp>(op);
858     if (resultNumber >= whileOp.getBeforeArguments().size())
861         whileOp.getBeforeArguments()[resultNumber].getType())
864     auto conditionOp = whileOp.getConditionOp();
865     BlockArgument conditionBbArg = whileOp.getBeforeArguments()[resultNumber];
870     auto yieldOp = whileOp.getYieldOp();
871     BlockArgument bodyBbArg = whileOp.getAfterArguments()[resultNumber];
909     auto whileOp = cast<scf::WhileOp>(op);
910     auto conditionOp = whileOp.getConditionOp();
915         whileOp.getBeforeArguments(), conditionOp.getArgs(), state);
917         whileOp.getAfterArguments(), whileOp.getYieldOp().getResults(), state);
946     auto whileOp = cast<scf::WhileOp>(op);
950     DenseSet<int64_t> indicesBefore = getTensorIndices(whileOp.getInits());
952         getTensorIndices(whileOp.getAfterArguments());
956         getBuffers(rewriter, whileOp.getInitsMutable(), options);
965       Value beforeArg = whileOp.getBeforeArguments()[it.index()];
979         llvm::map_range(whileOp.getAfterArguments(), [&](BlockArgument bbArg) {
991         whileOp.getLoc(), argsTypesAfter, castedInitArgs);
995                                           whileOp.getLoc());
997                                          whileOp.getLoc());
1009                              whileOp.getBeforeArguments(), indicesBefore);
1010     rewriter.mergeBlocks(whileOp.getBeforeBody(), newBeforeBody, newBeforeArgs);
1018                              whileOp.getAfterArguments(), indicesAfter);
1019     rewriter.mergeBlocks(whileOp.getAfterBody(), newAfterBody, newAfterArgs);
1030     auto whileOp = cast<scf::WhileOp>(op);
1036       if (bbArg.getOwner()->getParent() == &whileOp.getBefore()) {
1037         Value initArg = whileOp.getInits()[bbArg.getArgNumber()];
1038         auto yieldOp = whileOp.getYieldOp();
1052                &whileOp.getAfter()) {
1057     Value conditionYieldedVal = whileOp.getConditionOp().getArgs()[resultNum];
1078     auto whileOp = cast<scf::WhileOp>(op);
1084     auto conditionOp = whileOp.getConditionOp();
1097     auto yieldOp = whileOp.getYieldOp();
1172         } else if (auto whileOp =
1175               whileOp.getBeforeArguments()[it.index()], options);