Home
last modified time | relevance | path

Searched refs:getArgNumber (Results 1 – 25 of 40) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DOneShotModuleBufferize.cpp138 equivBbArgs[returnVal.getOperandNumber()] = bbArg.getArgNumber(); in annotateEquivalentReturnBbArg()
159 int64_t bbArgIdx = bbArg.getArgNumber(); in aliasingFuncOpBBArgsAnalysis()
187 funcOp.setArgAttr(bbArg.getArgNumber(), "bufferization.access", accessType); in annotateFuncArgAccess()
201 funcState.readBbArgs[funcOp].insert(bbArg.getArgNumber()); in funcOpBbArgReadWriteAnalysis()
202 funcState.writtenBbArgs[funcOp].insert(bbArg.getArgNumber()); in funcOpBbArgReadWriteAnalysis()
216 funcState.readBbArgs[funcOp].insert(bbArg.getArgNumber()); in funcOpBbArgReadWriteAnalysis()
218 funcState.writtenBbArgs[funcOp].insert(bbArg.getArgNumber()); in funcOpBbArgReadWriteAnalysis()
228 funcOp.removeArgAttr(bbArg.getArgNumber(), in removeBufferizationAttributes()
230 funcOp.removeArgAttr(bbArg.getArgNumber(), in removeBufferizationAttributes()
H A DDropEquivalentBufferResults.cpp86 resultToArgs[it.index()] = bbArg.getArgNumber(); in dropEquivalentBufferResults()
H A DBufferDeallocation.cpp332 Value sourceValue = operands[blockArg.getArgNumber()]; in introduceBlockArgCopy()
341 operands.slice(blockArg.getArgNumber(), 1).assign(*clone); in introduceBlockArgCopy()
H A DFuncBufferizableOpInterfaceImpl.cpp444 getBufferizedFunctionArgType(funcOp, bbArg.getArgNumber(), options); in bufferize()
505 bbArg.getArgNumber(), BufferizationDialect::kWritableAttrName)) in isWritable()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDetensorize.cpp111 int idx = blockArgument.getArgNumber(); in matchAndRewrite()
227 blockOperands.isOperandProduced(blockArgumentElem.getArgNumber())) in computeBranchOpDetensoring()
231 blockOperands.getOperandIndex(blockArgumentElem.getArgNumber())); in computeBranchOpDetensoring()
348 currentItemBlockArgument.getArgNumber())) in compute()
354 ownerBlockOperands[currentItemBlockArgument.getArgNumber()]); in compute()
423 blockOperands.isOperandProduced(blockArg.getArgNumber())) in compute()
427 blockOperands[blockArg.getArgNumber()].getDefiningOp(); in compute()
H A DHoisting.cpp245 bbArg.getArgNumber() != /*num iv=*/1 + yieldOperand.getOperandNumber()) in getLoopInvariantTransferWriteOpDefining()
283 assert(tensorBBArg.getArgNumber() >= /*numIvs=*/1); in hoistReadWrite()
284 unsigned initArgNumber = tensorBBArg.getArgNumber() - /*numIvs=*/1; in hoistReadWrite()
H A DConstantFold.cpp288 if (yieldArg.getArgNumber() != 0) in getRegionComputeFn()
H A DFusion.cpp395 tensor = *(forOp.getIterOperands().begin() + blockArg.getArgNumber()); in getProducerOfTensor()
H A DVectorization.cpp369 if (!arg || arg.getArgNumber() < linalgOp.getNumInputs()) in vectorizeOneOp()
374 arg.getArgNumber() - linalgOp.getNumInputs(), reductionOps); in vectorizeOneOp()
496 LDBG("new vectorized bbarg(" << bbarg.getArgNumber() << "): " << readValue); in vectorizeAsLinalgGeneric()
/llvm-project-15.0.7/mlir/lib/IR/
H A DBlock.cpp178 return insertArgument(it->getArgNumber(), type, loc); in insertArgument()
198 [&](BlockArgument arg) { return eraseIndices.test(arg.getArgNumber()); }); in eraseArguments()
208 unsigned index = firstDead->getArgNumber(); in eraseArguments()
/llvm-project-15.0.7/mlir/lib/Analysis/AliasAnalysis/
H A DLocalAliasAnalysis.cpp61 firstInputIndex = inputs[0].cast<BlockArgument>().getArgNumber(); in collectUnderlyingAddressValues()
62 lastInputIndex = inputs.back().cast<BlockArgument>().getArgNumber(); in collectUnderlyingAddressValues()
138 unsigned argNumber = arg.getArgNumber(); in collectUnderlyingAddressValues()
/llvm-project-15.0.7/mlir/test/lib/IR/
H A DTestPrintDefUse.cpp39 << blockArg.getArgNumber() << "\n"; in runOnOperation()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestAffineLoopParametricTiling.cpp60 if (blockArgument.getArgNumber() < nestDepth) { in getTilingParameters()
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DAsmParserState.cpp267 unsigned argIdx = blockArg.getArgNumber(); in addDefinition()
309 SMDefinition &argDef = blockDef.arguments[arg.getArgNumber()]; in addUses()
/llvm-project-15.0.7/mlir/test/lib/Analysis/
H A DTestAliasAnalysis.cpp34 llvm::errs() << "#" << arg.getArgNumber(); in printAliasOperand()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopCanonicalization.cpp95 if (!isShapePreserving(forOp, blockArg.getArgNumber() - 1)) in matchAndRewrite()
H A DLoopPipelining.cpp298 arg.getArgNumber() - 1); in createKernel()
/llvm-project-15.0.7/mlir/lib/Analysis/DataFlow/
H A DIntegerRangeAnalysis.cpp133 IntegerValueRangeLattice *lattice = argLattices[arg.getArgNumber()]; in visitNonControlFlowArguments()
H A DSparseAnalysis.cpp265 firstIndex = inputs.front().cast<BlockArgument>().getArgNumber(); in visitRegionSuccessors()
/llvm-project-15.0.7/mlir/lib/Transforms/
H A DViewOpGraph.cpp238 return "arg" + std::to_string(arg.getArgNumber()); in getLabel()
/llvm-project-15.0.7/mlir/lib/Analysis/
H A DLiveness.cpp284 os << "arg" << blockArg.getArgNumber() << "@" in print()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DValue.h312 unsigned getArgNumber() const { return getImpl()->index; } in getArgNumber() function
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp820 newBlockTransferArgs[newRegionIterArg.getArgNumber()] = castIn; in replaceTensorCastForOpIterArg()
830 newRegionIterArg.getArgNumber() - forOp.getNumInductionVars(); in replaceTensorCastForOpIterArg()
975 unsigned idx = bbArg.getArgNumber() - /*numIv=*/1; in matchAndRewrite()
1012 Value forOpResult = forOp.getResult(bbArg.getArgNumber() - /*iv=*/1); in matchAndRewrite()
2841 Value condOpArg = condOpArgs[yieldOpBlockArg.getArgNumber()]; in matchAndRewrite()
2874 Value condOpArg = condOpArgs[yieldOpBlockArg.getArgNumber()]; in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DArrayValueCopy.cpp248 collectArrayMentionFrom(rop.blockArgToSourceOp(ba.getArgNumber())); in collectArrayMentionFrom()
260 auto u = pred->getTerminator()->getOperand(ba.getArgNumber()); in collectArrayMentionFrom()
368 int64_t arg = blockArg.getArgNumber(); in arrayMentions()
/llvm-project-15.0.7/mlir/lib/Dialect/Transform/IR/
H A DTransformOps.cpp510 return (emitOpError() << "block argument #" << argument.getArgNumber()); in verify()

12