Home
last modified time | relevance | path

Searched defs:genericOp (Results 1 – 24 of 24) sorted by relevance

/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgDropUnitDims.cpp25 linalg::GenericOp genericOp) { in dropOutermostUnitDims() argument
56 __anon7924bc6f0302(linalg::GenericOp genericOp) runOnOperation() argument
58 for (auto genericOp : genericOps) { runOnOperation() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DInterchange.cpp36 interchangeGenericOpPrecondition(GenericOp genericOp, in interchangeGenericOpPrecondition() argument
50 mlir::linalg::interchangeGenericOp(RewriterBase &rewriter, GenericOp genericOp, in interchangeGenericOp() argument
[all...]
H A DSpecialize.cpp58 static bool areBinOpsSwapped(GenericOp genericOp) { in areBinOpsSwapped() argument
149 specializeLinalgContractions(RewriterBase & rewriter,GenericOp genericOp) specializeLinalgContractions() argument
263 specializeGenericOp(RewriterBase & rewriter,GenericOp genericOp) specializeGenericOp() argument
[all...]
H A DDataLayoutPropagation.cpp38 static bool hasGatherSemantics(linalg::GenericOp genericOp) { in hasGatherSemantics() argument
62 getPackingInfoFromOperand(OpOperand *opOperand, linalg::GenericOp genericOp, in getPackingInfoFromOperand() argument
228 GenericOp genericOp, OpOperand *opOperand) { in getOrCreatePackedViewOfOperand() argument
302 packGenericOp(RewriterBase & rewriter,GenericOp genericOp,Value dest,AffineMap packedOutIndexingMap,const PackInfo & packInfo) packGenericOp() argument
376 auto genericOp = packOp.getSource().getDefiningOp<GenericOp>(); bubbleUpPackOpThroughGenericOp() local
464 auto genericOp = matchAndRewrite() local
1002 getUnPackedOperand(GenericOp genericOp) getUnPackedOperand() argument
1052 pushDownUnPackOpThroughGenericOp(RewriterBase & rewriter,GenericOp genericOp,ControlPropagationFn controlFn) pushDownUnPackOpThroughGenericOp() argument
[all...]
H A DDecomposeLinalgOps.cpp146 DecomposeLinalgOp::createPeeledGenericOp(GenericOp genericOp, in createPeeledGenericOp()
212 DecomposeLinalgOp::createResidualGenericOp(GenericOp genericOp, in createResidualGenericOp()
250 DecomposeLinalgOp::matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
H A DGeneralization.cpp69 GenericOp genericOp = rewriter.create<GenericOp>( in generalizeNamedOp() local
H A DDetensorize.cpp56 GenericOp genericOp = dyn_cast_or_null<GenericOp>(op); in shouldBeDetensored() local
456 if (shouldBeDetensored(genericOp, typeConverte in compute() argument
374 if (auto genericOp = dyn_cast<GenericOp>(currentItemDefiningOp)) { compute() local
[all...]
H A DSplitReduction.cpp171 GenericOp genericOp = b.create<GenericOp>( in splitReduction() local
367 GenericOp genericOp = in splitReductionByScaling() local
H A DDropUnitDims.cpp233 replaceUnitDimIndexOps(GenericOp genericOp,const llvm::SmallDenseSet<unsigned> & unitDims,RewriterBase & rewriter) replaceUnitDimIndexOps() argument
341 dropUnitExtentFromOperandMetadata(MLIRContext * context,GenericOp genericOp,OpOperand * opOperand,llvm::SmallDenseMap<unsigned,unsigned> & oldDimsToNewDimsMap,ArrayRef<AffineExpr> dimReplacements) dropUnitExtentFromOperandMetadata() argument
389 dropUnitDims(RewriterBase & rewriter,GenericOp genericOp,const ControlDropUnitDims & options) dropUnitDims() argument
[all...]
H A DConvertToDestinationStyle.cpp142 auto genericOp = rewriter.create<linalg::GenericOp>( in movePaddingToFillOrGenericOp() local
401 auto genericOp = rewriter.create<linalg::GenericOp>( rewriteInDestinationPassingStyle() local
H A DBlockPackMatmul.cpp185 auto genericOp = in blockPackMatmul() local
H A DConvertConv2DToImg2Col.cpp191 auto genericOp = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
475 auto genericOp = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
610 auto genericOp = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
H A DEraseUnusedOperandsAndResults.cpp17 static bool isResultValueDead(linalg::GenericOp genericOp, OpResult result) { in isResultValueDead()
H A DElementwiseOpFusion.cpp424 matchAndRewrite(GenericOp genericOp,PatternRewriter & rewriter) const matchAndRewrite() argument
1215 getCollapsableIterationSpaceDims(GenericOp genericOp,OpOperand * fusableOperand,ArrayRef<ReassociationIndices> reassociation) getCollapsableIterationSpaceDims() argument
1621 if (GenericOp genericOp = dyn_cast<GenericOp>(op.getOperation())) { createCollapsedOp() local
1735 matchAndRewrite(GenericOp genericOp,PatternRewriter & rewriter) const matchAndRewrite() argument
1898 matchAndRewrite(GenericOp genericOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...]
H A DTilingInterfaceImpl.cpp447 auto genericOp = tileToPartialReduction() local
H A DTransforms.cpp826 if (!genericOp) { in packMatmulGreedily() local
[all...]
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DExpandOps.cpp57 auto genericOp = rewriter.create<memref::GenericAtomicRMWOp>( in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp76 isaFillOpInterface(GenericOp genericOp) isaFillOpInterface() argument
106 isaElemwiseSingleUnaryOrBinaryOpInterface(linalg::GenericOp genericOp,unsigned arity) isaElemwiseSingleUnaryOrBinaryOpInterface() argument
142 isaElemwiseSingleUnaryOpInterface(linalg::GenericOp genericOp) isaElemwiseSingleUnaryOpInterface() argument
153 isaElemwiseSingleBinaryOpInterface(linalg::GenericOp genericOp) isaElemwiseSingleBinaryOpInterface() argument
H A DLinalgOps.cpp2624 auto genericOp = builder.create<linalg::GenericOp>( reduce() local
2647 auto genericOp = builder.create<linalg::GenericOp>( buildSubAndExpOp() local
2674 auto genericOp = builder.create<linalg::GenericOp>( buildDivOp() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DIterationGraphSorter.cpp127 IterationGraphSorter::fromGenericOp(linalg::GenericOp genericOp) { in fromGenericOp()
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp1552 auto genericOp = b.create<linalg::GenericOp>( matchAndRewrite() local
1922 auto genericOp = rewriter.create<linalg::GenericOp>( matchAndRewrite() local
2104 auto genericOp = rewriter.create<linalg::GenericOp>( matchAndRewrite() local
2178 auto genericOp = rewriter.create<linalg::GenericOp>( matchAndRewrite() local
[all...]
H A DTosaToLinalgNamed.cpp880 auto genericOp = rewriter.create<linalg::GenericOp>( matchAndRewrite() local
/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h481 if (auto genericOp = dyn_cast_or_null<GenericOp>(op)) { member
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp630 auto genericOp = dyn_cast<linalg::GenericOp>(producerOp); replaceForAllWithNewSignature() local