Home
last modified time | relevance | path

Searched refs:resultNumber (Results 1 – 15 of 15) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTilingInterfaceImpl.cpp95 getResultTilePosition(Operation *op, OpBuilder &b, unsigned resultNumber, in getResultTilePosition()
122 OpOperand *outOperand = linalgOp.getOutputOperand(resultNumber); in getResultTilePosition()
137 unsigned resultNumber, in generateResultTileValue()
149 linalgOp.getTiedIndexingMapForResult(op->getResult(resultNumber)); in generateResultTileValue()
180 return tiledOp[0]->getResult(resultNumber); in generateResultTileValue()
H A DFusePadOpWithLinalgProducer.cpp92 unsigned resultNumber = source.cast<OpResult>().getResultNumber(); in matchAndRewrite() local
113 clonedOp.setOutputOperand(resultNumber, slice.getResult()); in matchAndRewrite()
117 padOp, clonedOp.getResult(resultNumber), fillTensor.getResult(0), in matchAndRewrite()
H A DElementwiseOpFusion.cpp794 int64_t resultNumber = opResult.getResultNumber(); in fuseWithReshapeByExpansion() local
795 if (!isExpanding && resultTypes[resultNumber] != opResult.getType()) { in fuseWithReshapeByExpansion()
799 genericOp.getOutputOperand(resultNumber)), in fuseWithReshapeByExpansion()
803 fusedOp->getResult(resultNumber), reassociation)); in fuseWithReshapeByExpansion()
805 resultVals.push_back(fusedOp->getResult(resultNumber)); in fuseWithReshapeByExpansion()
H A DTransforms.cpp297 int64_t resultNumber = en.index(); in rewriteAsPaddedOp() local
301 for (Value v : reifiedResultShapes[resultNumber]) in rewriteAsPaddedOp()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp93 unsigned resultNumber = dimValue.getResultNumber(); in matchAndRewrite() local
95 if (reifiedResultShapes[resultNumber].size() != in matchAndRewrite()
99 rewriter.replaceOp(dimOp, reifiedResultShapes[resultNumber][*dimIndex]); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DOperation.h723 detail::OutOfLineOpResult *getOutOfLineOpResult(unsigned resultNumber) { in getOutOfLineOpResult() argument
728 ++resultNumber; in getOutOfLineOpResult()
732 detail::InlineOpResult *getInlineOpResult(unsigned resultNumber) { in getInlineOpResult() argument
735 return reinterpret_cast<detail::InlineOpResult *>(this) - ++resultNumber; in getInlineOpResult()
740 detail::OpResultImpl *getOpResultImpl(unsigned resultNumber) { in getOpResultImpl() argument
742 if (resultNumber < maxInlineResults) in getOpResultImpl()
743 return getInlineOpResult(resultNumber); in getOpResultImpl()
744 return getOutOfLineOpResult(resultNumber - maxInlineResults); in getOpResultImpl()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopCanonicalization.cpp138 unsigned resultNumber = opResult.getResultNumber(); in matchAndRewrite() local
139 if (!isShapePreserving(forOp, resultNumber)) in matchAndRewrite()
142 dimOp.getSourceMutable().assign(forOp.getIterOperands()[resultNumber]); in matchAndRewrite()
H A DBufferizableOpInterfaceImpl.cpp593 unsigned int resultNumber = opResult.getResultNumber(); in bufferRelation() local
597 if (resultNumber >= whileOp.getBeforeArguments().size()) in bufferRelation()
600 whileOp.getBeforeArguments()[resultNumber].getType()) in bufferRelation()
604 BlockArgument conditionBbArg = whileOp.getBeforeArguments()[resultNumber]; in bufferRelation()
605 Value conditionOperand = conditionOp.getArgs()[resultNumber]; in bufferRelation()
610 BlockArgument bodyBbArg = whileOp.getAfterArguments()[resultNumber]; in bufferRelation()
611 Value yieldOperand = yieldOp.getOperand(resultNumber); in bufferRelation()
H A DTileUsingInterface.cpp483 unsigned resultNumber = uses.get().cast<OpResult>().getResultNumber(); in returningMatchAndRewrite() local
486 operandNumber, unfusedProducerOpDestValues[resultNumber]); in returningMatchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Transform/IR/
H A DTransformInterfaces.cpp286 transform::TransformResults::get(unsigned resultNumber) const { in get()
287 assert(resultNumber < segments.size() && in get()
289 assert(segments[resultNumber].data() != nullptr && "querying unset results"); in get()
290 return segments[resultNumber]; in get()
/llvm-project-15.0.7/mlir/lib/Tools/mlir-lsp-server/
H A DMLIRServer.cpp143 unsigned resultNumber = 0; in getResultNumberFromLoc() local
144 return numberStr.consumeInteger(10, resultNumber) ? Optional<unsigned>() in getResultNumberFromLoc()
145 : resultNumber; in getResultNumberFromLoc()
521 if (Optional<unsigned> resultNumber = getResultNumberFromLoc(posLoc)) { in buildHoverForOperationResult() local
522 if ((resultStart + *resultNumber) < resultEnd) { in buildHoverForOperationResult()
523 resultStart += *resultNumber; in buildHoverForOperationResult()
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DTilingInterface.td114 "unsigned":$resultNumber,
155 "unsigned":$resultNumber,
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1731 unsigned resultNumber = resultValue.getResultNumber(); in matchAndRewrite() local
1738 OpOperand *outOperand = linalgOp.getOutputOperand(resultNumber); in matchAndRewrite()
1743 outputOperands[resultNumber] = newOperand; in matchAndRewrite()
1748 resultTypes[resultNumber] = resultType; in matchAndRewrite()
1753 loc, resultValue.getType(), newOp->getResult(resultNumber)); in matchAndRewrite()
1756 results[resultNumber] = castBack; in matchAndRewrite()
1758 rewriter.replaceOp(castOp, newOp->getResult(resultNumber)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp286 unsigned resultNumber = use->get().cast<OpResult>().getResultNumber(); in cloneAndFuse() local
291 use->getOwner(), [&] { use->set(cloned->getOpResult(resultNumber)); }); in cloneAndFuse()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Transform/IR/
H A DTransformInterfaces.h548 ArrayRef<Operation *> get(unsigned resultNumber) const;