Lines Matching defs:bbArg

24 // * `funcOpBbArgReadWriteAnalysis` determines whether or not a tensor bbArg is
93                                           BlockArgument bbArg) {
106   equivBbArgs[returnVal.getOperandNumber()] = bbArg.getArgNumber();
119     // return value may alias with any tensor bbArg.
140   for (BlockArgument bbArg : funcOp.getArguments()) {
141     if (isa<RankedTensorType>(bbArg.getType())) {
142       int64_t bbArgIdx = bbArg.getArgNumber();
149             if (state.areAliasingBufferizedValues(returnVal.get(), bbArg))
168     for (BlockArgument bbArg : funcOp.getArguments()) {
169       if (isa<RankedTensorType>(bbArg.getType())) {
170         if (state.areEquivalentBufferizedValues(v, bbArg)) {
172             annotateEquivalentReturnBbArg(opOperand, bbArg);
173           return bbArg.getArgNumber();
256       BlockArgument bbArg = funcOp.getArgument(idx);
257       isRead = state.isValueRead(bbArg);
258       isWritten = state.isValueWritten(bbArg);
274 static void removeBufferizationAttributes(BlockArgument bbArg) {
275   auto funcOp = cast<func::FuncOp>(bbArg.getOwner()->getParentOp());
276   funcOp.removeArgAttr(bbArg.getArgNumber(),
278   funcOp.removeArgAttr(bbArg.getArgNumber(),
537     for (BlockArgument bbArg : op.getArguments())
538       removeBufferizationAttributes(bbArg);