125                                           BlockArgument bbArg) {  in annotateEquivalentReturnBbArg()  argument
138   equivBbArgs[returnVal.getOperandNumber()] = bbArg.getArgNumber();  in annotateEquivalentReturnBbArg()
156       for (BlockArgument bbArg : funcOp.getArguments())  in aliasingFuncOpBBArgsAnalysis()  local
157         if (bbArg.getType().isa<RankedTensorType>()) {  in aliasingFuncOpBBArgsAnalysis()
159           int64_t bbArgIdx = bbArg.getArgNumber();  in aliasingFuncOpBBArgsAnalysis()
160           if (state.areEquivalentBufferizedValues(returnVal.get(), bbArg)) {  in aliasingFuncOpBBArgsAnalysis()
163               annotateEquivalentReturnBbArg(returnVal, bbArg);  in aliasingFuncOpBBArgsAnalysis()
165           if (state.areAliasingBufferizedValues(returnVal.get(), bbArg)) {  in aliasingFuncOpBBArgsAnalysis()
174 static void annotateFuncArgAccess(func::FuncOp funcOp, BlockArgument bbArg,  in annotateFuncArgAccess()  argument
187   funcOp.setArgAttr(bbArg.getArgNumber(), "bufferization.access", accessType);  in annotateFuncArgAccess()
200     for (BlockArgument bbArg : funcOp.getArguments()) {  in funcOpBbArgReadWriteAnalysis()  local
201       funcState.readBbArgs[funcOp].insert(bbArg.getArgNumber());  in funcOpBbArgReadWriteAnalysis()
202       funcState.writtenBbArgs[funcOp].insert(bbArg.getArgNumber());  in funcOpBbArgReadWriteAnalysis()
208   for (BlockArgument bbArg : funcOp.getArguments()) {  in funcOpBbArgReadWriteAnalysis()  local
209     if (!bbArg.getType().isa<TensorType>())  in funcOpBbArgReadWriteAnalysis()
211     bool isRead = state.isValueRead(bbArg);  in funcOpBbArgReadWriteAnalysis()
212     bool isWritten = state.isValueWritten(bbArg);  in funcOpBbArgReadWriteAnalysis()
214       annotateFuncArgAccess(funcOp, bbArg, isRead, isWritten);  in funcOpBbArgReadWriteAnalysis()
216       funcState.readBbArgs[funcOp].insert(bbArg.getArgNumber());  in funcOpBbArgReadWriteAnalysis()
218       funcState.writtenBbArgs[funcOp].insert(bbArg.getArgNumber());  in funcOpBbArgReadWriteAnalysis()
226 static void removeBufferizationAttributes(BlockArgument bbArg) {  in removeBufferizationAttributes()  argument
227   auto funcOp = cast<func::FuncOp>(bbArg.getOwner()->getParentOp());  in removeBufferizationAttributes()
228   funcOp.removeArgAttr(bbArg.getArgNumber(),  in removeBufferizationAttributes()
230   funcOp.removeArgAttr(bbArg.getArgNumber(),  in removeBufferizationAttributes()
434     for (BlockArgument bbArg : op.getArguments())  in bufferizeModuleOp()  local
435       removeBufferizationAttributes(bbArg);  in bufferizeModuleOp()