Lines Matching refs:bbArg
108 for (auto bbArg : b.getArguments()) in BufferizationAliasInfo() local
109 if (bbArg.getType().isa<TensorType>()) in BufferizationAliasInfo()
110 createAliasInfoEntry(bbArg); in BufferizationAliasInfo()
243 if (auto bbArg = v.dyn_cast<BlockArgument>()) { in gatherYieldedTensors() local
244 if (bbArg.getOwner()->getParentOp() == returnOp->getParentOp()) in gatherYieldedTensors()
245 yieldedTensors.insert(bbArg); in gatherYieldedTensors()
315 if (auto bbArg = value.dyn_cast<BlockArgument>()) in isWritable() local
317 getOptions().dynCastBufferizableOp(bbArg.getOwner()->getParentOp())) in isWritable()
318 return bufferizableOp.isWritable(bbArg, *this); in isWritable()
407 auto bbArg = lastWrite.cast<BlockArgument>(); in annotateConflict() local
409 id + "[LAST-WRITE: bbArg " + std::to_string(bbArg.getArgNumber()) + "]"; in annotateConflict()
410 bbArg.getOwner()->getParentOp()->setAttr(lastWriteAttr, b.getUnitAttr()); in annotateConflict()
552 auto bbArg = lastWrite.cast<BlockArgument>(); in hasReadAfterWriteInterference() local
553 Block *block = bbArg.getOwner(); in hasReadAfterWriteInterference()
913 if (auto bbArg = equivVal.dyn_cast<BlockArgument>()) { in assertDestinationPassingStyle() local
914 Operation *definingOp = bbArg.getOwner()->getParentOp(); in assertDestinationPassingStyle()