Home
last modified time | relevance | path

Searched defs:opOperand (Results 1 – 25 of 27) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DPadding.cpp28 OpOperand *opOperand, in computePaddedShape() argument
103 RewriterBase &rewriter, linalg::LinalgOp opToPad, OpOperand *opOperand, in padOperandToSmallestStaticBoundingBox() argument
287 OpOperand &opOperand = paddedOp->getOpOperand(en.index()); padAndHoistLinalgOp() local
[all...]
H A DInlineScalarOperands.cpp74 OpOperand *opOperand = genericOp.getDpsInputOperand(idx); in matchAndRewrite() local
[all...]
H A DBufferizableOpInterfaceImpl.cpp61 OpOperand *opOperand = op.getDpsInitOperand(opResult.getResultNumber()); in bufferizeDestinationStyleOpInterface() local
H A DSplitReduction.cpp215 static AffineMap scaleReductionDim(LinalgOp op, OpOperand &opOperand, in scaleReductionDim()
230 static AffineMap insertParallelDim(LinalgOp op, OpOperand &opOperand, in insertParallelDim()
H A DDataLayoutPropagation.cpp62 getPackingInfoFromOperand(OpOperand *opOperand, linalg::GenericOp genericOp, in getPackingInfoFromOperand() argument
228 GenericOp genericOp, OpOperand *opOperand) { in getOrCreatePackedViewOfOperand() argument
432 OpOperand *opOperand in bubbleUpPackOpThroughGenericOp() local
[all...]
H A DFusion.cpp268 OpOperand *opOperand = in fuseProducerOfTensor() local
H A DDetensorize.cpp58 llvm::all_of(genericOp->getOpOperands(), [&](OpOperand &opOperand) { in shouldBeDetensored() argument
H A DElementwiseOpFusion.cpp351 for (const auto &opOperand : llvm::enumerate(producer.getDpsInitsMutable())) { fuseElementwiseOps() local
1478 getCollapsedOpOperand(Location loc,LinalgOp op,OpOperand * opOperand,const CollapsingInfo & collapsingInfo,OpBuilder & builder) getCollapsedOpOperand() argument
1551 __anona62ab6fa1902(OpOperand *opOperand) collapseOperandsAndResults() argument
[all...]
H A DTransforms.cpp634 transposeOneLinalgOperandAndReplace(RewriterBase & rewriter,LinalgOp linalgOp,OpOperand & opOperand,ArrayRef<int64_t> permutation,Value transposedValue) transposeOneLinalgOperandAndReplace() argument
739 OpOperand &opOperand = packTranspose() local
[all...]
H A DDropUnitDims.cpp341 dropUnitExtentFromOperandMetadata(MLIRContext * context,GenericOp genericOp,OpOperand * opOperand,llvm::SmallDenseMap<unsigned,unsigned> & oldDimsToNewDimsMap,ArrayRef<AffineExpr> dimReplacements) dropUnitExtentFromOperandMetadata() argument
[all...]
/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestDataLayoutPropagation.cpp36 patterns, [](OpOperand *opOperand) { return true; }); in runOnOperation() argument
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp205 bool AllocTensorOp::bufferizesToMemoryRead(OpOperand &opOperand, in bufferizesToMemoryRead() argument
212 bool AllocTensorOp::bufferizesToMemoryWrite(OpOperand &opOperand, in bufferizesToMemoryWrite() argument
219 AliasingValueList AllocTensorOp::getAliasingValues(OpOperand &opOperand, in getAliasingValues() argument
549 return opOperand == getSourceMutable(); in bufferizesToMemoryRead() argument
554 if (opOperand == getDestMutable()) { in bufferizesToMemoryWrite() argument
563 mustBufferizeInPlace(OpOperand & opOperand,const AnalysisState & state) mustBufferizeInPlace() argument
571 getAliasingValues(OpOperand & opOperand,const AnalysisState & state) getAliasingValues() argument
[all...]
H A DBufferizableOpInterface.cpp978 unknownGetAliasingValues(OpOperand & opOperand) unknownGetAliasingValues() argument
[all...]
/llvm-project/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.h45 : opOperand(opOperand), relation(relation), isDefinite(isDefinite) {} in opOperand() function
47 OpOperand *opOperand; member
[all...]
/llvm-project/mlir/lib/Dialect/Mesh/Interfaces/
H A DShardingInterface.cpp143 getMeshShardingAttr(OpOperand & opOperand) getMeshShardingAttr() argument
435 getShardingAttribute(OpOperand & opOperand,const ShardingOption & shardingOption,AffineMap map) getShardingAttribute() argument
518 addShardOp(OpBuilder & b,OpOperand & opOperand,const ShardingOption & shardingOption,AffineMap map) addShardOp() argument
[all...]
/llvm-project/mlir/lib/CAPI/IR/
H A DIR.cpp1015 return wrap(opOperand); in mlirOpOperandIsNull() argument
1002 OpOperand *opOperand = cppValue.use_begin().getOperand(); mlirValueGetFirstUse() local
1017 mlirOpOperandGetOwner(MlirOpOperand opOperand) mlirOpOperandGetOwner() argument
1021 mlirOpOperandGetValue(MlirOpOperand opOperand) mlirOpOperandGetValue() argument
1025 mlirOpOperandGetOperandNumber(MlirOpOperand opOperand) mlirOpOperandGetOperandNumber() argument
1029 mlirOpOperandGetNextUse(MlirOpOperand opOperand) mlirOpOperandGetNextUse() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgMatchOps.cpp97 getBody()->getTerminator()->getOpOperands(), [&](OpOperand &opOperand) { in matchOperation() argument
101 filtered, [](OpOperand &opOperand) { return opOperand.get(); })); in matchOperation() argument
/llvm-project/mlir/lib/Analysis/DataFlow/
H A DSparseAnalysis.cpp470 OpOperand &opOperand = op->getOpOperand(index); visitOperation() local
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DOneShotAnalysis.cpp90 static void setInPlaceOpOperand(OpOperand &opOperand, bool inPlace) { in setInPlaceOpOperand() argument
252 isInplaceMemoryWrite(OpOperand & opOperand,const OneShotAnalysisState & state) isInplaceMemoryWrite() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp39 for (auto &opOperand : linalgOp->getOpOperands()) { canOpOperandsBeDroppedImpl() local
1120 getIndexingMapIndex(OpOperand * opOperand) getIndexingMapIndex() argument
[all...]
/llvm-project/mlir/lib/Bindings/Python/
H A DIRCore.cpp543 PyOpOperand(MlirOpOperand opOperand) PyOpOperand() argument
562 MlirOpOperand opOperand; global() member in __anon0d8b16ee0311::PyOpOperand
567 PyOpOperandIterator(MlirOpOperand opOperand) PyOpOperandIterator() argument
587 MlirOpOperand opOperand; global() member in __anon0d8b16ee0311::PyOpOperandIterator
[all...]
/llvm-project/mlir/lib/IR/
H A DOperationSupport.cpp441 MutableOperandRange(OpOperand & opOperand) MutableOperandRange() argument
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp744 __anon42121b781302(OpOperand &opOperand) getTensorOutputTypes() argument
[all...]
/llvm-project/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp214 __anon59fc1c1d0302(OpOperand &opOperand) outlineSingleBlockRegion() argument
/llvm-project/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp1619 newOuts.push_back(opOperand->get()); in matchAndRewrite() local
1640 OpOperand *opOperand = forallOp.getTiedOpOperand(result); matchAndRewrite() local

12