Lines Matching defs:bbArg

165     if (auto bbArg = dyn_cast<BlockArgument>(value))
166       return getAliasingBranchOpOperands(op, bbArg, state);
448 /// bbArg/yielded value pairs who's buffer relation is "Equivalent".
607     // its matching bbArg may.
631     BlockArgument bbArg = forOp.getTiedLoopRegionIterArg(opResult);
633         bbArg, forOp.getTiedLoopYieldedValue(bbArg)->get());
640     // Interestingly, scf::ForOp's bbArg can **always** be viewed
643     //      alloc + optional copy makes the bbArg itself inplaceable.
644     //   2. Or the matching iter operand is bufferized inplace and bbArg just
709       BlockArgument bbArg = forOp.getTiedLoopRegionIterArg(opResult);
710       return bufferization::getBufferType(bbArg, options, invocationStack);
714     BlockArgument bbArg = cast<BlockArgument>(value);
715     unsigned resultNum = forOp.getTiedLoopResult(bbArg).getResultNumber();
807                << " is not equivalent to the corresponding iter bbArg";
882     // Interestingly, scf::WhileOp's bbArg can **always** be viewed
885     //      alloc + optional copy makes the bbArg itself inplaceable.
886     //   2. Or the matching iter operand is bufferized inplace and bbArg just
913     // bbArg?
977     // The result types of a WhileOp are the same as the "after" bbArg types.
979         llvm::map_range(whileOp.getAfterArguments(), [&](BlockArgument bbArg) {
980           if (!isa<TensorType>(bbArg.getType()))
981             return bbArg.getType();
984               *bufferization::getBufferType(bbArg, options));
1035     if (auto bbArg = dyn_cast<BlockArgument>(value)) {
1036       if (bbArg.getOwner()->getParent() == &whileOp.getBefore()) {
1037         Value initArg = whileOp.getInits()[bbArg.getArgNumber()];
1039         Value yieldedValue = yieldOp.getOperand(bbArg.getArgNumber());
1041             op, bbArg, initArg, yieldedValue, options, invocationStack);
1046     // The bufferized "after" bbArg type can be directly computed from the
1047     // bufferized "before" bbArg type.
1094                << " is not equivalent to the corresponding iter bbArg";
1107                << " is not equivalent to the corresponding iter bbArg";
1221     // uses of its matching bbArg may.
1262       BlockArgument bbArg = std::get<0>(it);
1265           forallOp.getLoc(), bbArg.getType(), buffer);
1266       bbArg.replaceAllUsesWith(bufferAsTensor);
1302     if (auto bbArg = dyn_cast<BlockArgument>(value))
1306           forallOp.getTiedOpOperand(bbArg)->get(), options, invocationStack);