Searched refs:bbArgs (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | FusionOnTensors.cpp | 222 SmallVector<BlockArgument> bbArgs; in getTiedBBArgs() local
228 bbArgs.push_back(bbArg); in getTiedBBArgs()
234 return {bbArgs.rbegin(), bbArgs.rend()}; in getTiedBBArgs()
239 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in getTiedIterArg() local
240 if (bbArgs.size() != tileLoopOps.size()) in getTiedIterArg()
242 return &tileLoopOps.front().getOpOperandForRegionIterArg(bbArgs.front()); in getTiedIterArg()
269 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in hasOtherUses() local
270 return !all_of(bbArgs, [&](BlockArgument bbArg) { in hasOtherUses()
271 return bbArg.hasOneUse() || bbArg == bbArgs.back(); in hasOtherUses()
|
| H A D | SplitReduction.cpp | 430 [combinerOp](OpBuilder &b, Location loc, ValueRange bbArgs) { in splitReductionByScaling() argument
432 clonedReductionOp->setOperand(0, bbArgs[0]); in splitReductionByScaling()
433 clonedReductionOp->setOperand(1, bbArgs[1]); in splitReductionByScaling()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 252 DenseSet<int64_t> getEquivalentBuffers(Block::BlockArgListType bbArgs, in getEquivalentBuffers() argument
255 unsigned int minSize = std::min(bbArgs.size(), yieldedValues.size()); in getEquivalentBuffers()
258 if (!bbArgs[i].getType().isa<TensorType>() || in getEquivalentBuffers()
261 if (state.areEquivalentBufferizedValues(bbArgs[i], yieldedValues[i])) in getEquivalentBuffers()
357 getBbArgReplacements(RewriterBase &rewriter, Block::BlockArgListType bbArgs, in getBbArgReplacements() argument
360 for (const auto &it : llvm::enumerate(bbArgs)) { in getBbArgReplacements()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 77 ArrayRef<BlockArgument> bbArgs = loopBody->getArguments(); in replaceLoopWithNewYields() local
79 llvm::zip(bbArgs, newLoopBody->getArguments().take_front(bbArgs.size()))) in replaceLoopWithNewYields()
|
| /llvm-project-15.0.7/mlir/test/Dialect/SCF/ |
| H A D | one-shot-bufferize-analysis.mlir | 51 // scf.for bbArgs are always inplaceable seen from ops inside the body:
|
| /llvm-project-15.0.7/mlir/docs/ |
| H A D | Bufferization.md | 423 I.e., `CallOp`, `ReturnOp` and function bbArgs are not bufferizable. Users can
|