| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | OneShotModuleBufferize.cpp | 125 BlockArgument bbArg) { in annotateEquivalentReturnBbArg()
156 for (BlockArgument bbArg : funcOp.getArguments()) in aliasingFuncOpBBArgsAnalysis() local
174 static void annotateFuncArgAccess(func::FuncOp funcOp, BlockArgument bbArg, in annotateFuncArgAccess()
200 for (BlockArgument bbArg : funcOp.getArguments()) { in funcOpBbArgReadWriteAnalysis() local
208 for (BlockArgument bbArg : funcOp.getArguments()) { in funcOpBbArgReadWriteAnalysis() local
226 static void removeBufferizationAttributes(BlockArgument bbArg) { in removeBufferizationAttributes()
434 for (BlockArgument bbArg : op.getArguments()) in bufferizeModuleOp() local
|
| H A D | AllocTensorElimination.cpp | 30 if (auto bbArg = val.dyn_cast<BlockArgument>()) { in neededValuesDominateInsertionPoint() local
72 if (auto bbArg = val.dyn_cast<BlockArgument>()) { in findValidInsertionPoint() local
|
| H A D | OneShotAnalysis.cpp | 108 for (auto bbArg : b.getArguments()) in BufferizationAliasInfo() local
243 if (auto bbArg = v.dyn_cast<BlockArgument>()) { in gatherYieldedTensors() local
315 if (auto bbArg = value.dyn_cast<BlockArgument>()) in isWritable() local
407 auto bbArg = lastWrite.cast<BlockArgument>(); in annotateConflict() local
552 auto bbArg = lastWrite.cast<BlockArgument>(); in hasReadAfterWriteInterference() local
913 if (auto bbArg = equivVal.dyn_cast<BlockArgument>()) { in assertDestinationPassingStyle() local
|
| H A D | DropEquivalentBufferResults.cpp | 80 for (BlockArgument bbArg : funcOp.getArguments()) { in dropEquivalentBufferResults() local
|
| H A D | FuncBufferizableOpInterfaceImpl.cpp | 499 BlockArgument bbArg = value.dyn_cast<BlockArgument>(); in isWritable() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | FusionOnTensors.cpp | 220 SmallVector<BlockArgument> TileLoopNest::getTiedBBArgs(BlockArgument bbArg) { in getTiedBBArgs()
237 OpOperand *TileLoopNest::getTiedIterArg(BlockArgument bbArg) { in getTiedIterArg()
245 bool TileLoopNest::hasOtherUses(BlockArgument bbArg, in hasOtherUses()
270 return !all_of(bbArgs, [&](BlockArgument bbArg) { in hasOtherUses()
355 BlockArgument bbArg = consumerOp.getTiedBlockArgument(consumerOpOperand); in fuseProducer() local
362 if (auto bbArg = sliceOp.getSource().dyn_cast<BlockArgument>()) { in fuseProducer() local
|
| H A D | ElementwiseOpFusion.cpp | 178 for (BlockArgument bbArg : consumerBlock.getArguments().take_front( in generateFusedElementwiseOpRegion() local
186 for (BlockArgument bbArg : in generateFusedElementwiseOpRegion() local
194 BlockArgument bbArg = producerBlock.getArguments() in generateFusedElementwiseOpRegion() local
201 for (BlockArgument bbArg : in generateFusedElementwiseOpRegion() local
207 for (BlockArgument bbArg : in generateFusedElementwiseOpRegion() local
|
| H A D | Hoisting.cpp | 243 auto bbArg = insertSliceOp.getDest().dyn_cast<BlockArgument>(); in getLoopInvariantTransferWriteOpDefining() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.cpp | 310 if (auto bbArg = value.dyn_cast<BlockArgument>()) { in setInsertionPointAfter() local
582 if (auto bbArg = value.dyn_cast<BlockArgument>()) in getBufferType() local
691 auto bbArg = value.dyn_cast<BlockArgument>(); in isFunctionArgument() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 405 auto bbArg = forOp.getRegionIterArgForOpOperand(forOperand); in bufferRelation() local
738 llvm::map_range(whileOp.getAfterArguments(), [&](BlockArgument bbArg) { in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorDistribute.cpp | 48 Value bbArg = warpOpBody->getArgument(it.index()); in rewriteWarpOpToScfFor() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 974 for (BlockArgument bbArg : forOp.getRegionIterArgs()) { in matchAndRewrite() local
|