Lines Matching refs:GenericOp
68 static bool areElementwiseOpsFusable(GenericOp producer, GenericOp consumer, in areElementwiseOpsFusable()
139 generateFusedElementwiseOpRegion(PatternRewriter &rewriter, GenericOp fusedOp, in generateFusedElementwiseOpRegion()
143 auto producer = cast<GenericOp>(consumerOpOperand->get().getDefiningOp()); in generateFusedElementwiseOpRegion()
144 auto consumer = cast<GenericOp>(consumerOpOperand->getOwner()); in generateFusedElementwiseOpRegion()
250 fuseElementwiseOpsImpl(GenericOp producer, OpOperand *consumerOpOperand, in fuseElementwiseOpsImpl()
253 auto consumer = cast<GenericOp>(consumerOpOperand->getOwner()); in fuseElementwiseOpsImpl()
318 auto fusedOp = rewriter.create<GenericOp>( in fuseElementwiseOpsImpl()
355 GenericOp producer, const ControlFusionFn &controlFn) { in fuseElementwiseOps()
365 class FuseElementwiseOps : public OpRewritePattern<GenericOp> {
369 : OpRewritePattern<GenericOp>(context, benefit), in FuseElementwiseOps()
372 LogicalResult matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
377 dyn_cast_or_null<GenericOp>(opOperand->get().getDefiningOp()); in matchAndRewrite()
453 static bool isFusableWithReshapeByDimExpansion(GenericOp genericOp, in isFusableWithReshapeByDimExpansion()
563 static LogicalResult isGenericOpExpandable(GenericOp genericOp, in isGenericOpExpandable()
684 fuseWithReshapeByExpansion(GenericOp genericOp, Operation *reshapeOp, in fuseWithReshapeByExpansion()
780 rewriter.create<GenericOp>(genericOp.getLoc(), resultTypes, in fuseWithReshapeByExpansion()
818 : public OpRewritePattern<GenericOp> {
823 : OpRewritePattern<GenericOp>(context, benefit), in FoldWithProducerReshapeOpByExpansion()
826 LogicalResult matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
868 GenericOp producer = reshapeOp.getSrc().getDefiningOp<GenericOp>(); in matchAndRewrite()
1005 getCollapsableIterationSpaceDims(GenericOp genericOp, OpOperand *fusableOperand, in getCollapsableIterationSpaceDims()
1271 static Value getCollapsedOpOperand(Location loc, GenericOp genericOp, in getCollapsedOpOperand()
1333 GenericOp genericOp, ArrayRef<ReassociationIndices> foldedIterationDims, in collapseGenericOpIterationDims()
1381 auto collapsedGenericOp = rewriter.create<linalg::GenericOp>( in collapseGenericOpIterationDims()
1438 : public OpRewritePattern<GenericOp> {
1443 : OpRewritePattern<GenericOp>(context, benefit), in FoldWithProducerReshapeOpByCollapsing()
1446 LogicalResult matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
1488 class FoldScalarOrSplatConstant : public OpRewritePattern<GenericOp> {
1491 : OpRewritePattern<GenericOp>(context, benefit) {} in FoldScalarOrSplatConstant()
1493 LogicalResult matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
1562 auto fusedOp = rewriter.create<GenericOp>( in matchAndRewrite()
1599 struct RemoveOutsDependency : public OpRewritePattern<GenericOp> {
1600 using OpRewritePattern<GenericOp>::OpRewritePattern;
1602 LogicalResult matchAndRewrite(GenericOp op, in matchAndRewrite()
1646 struct FoldFillWithGenericOp : public OpRewritePattern<GenericOp> {
1647 using OpRewritePattern<GenericOp>::OpRewritePattern;
1649 LogicalResult matchAndRewrite(GenericOp genericOp, in matchAndRewrite()
1726 GenericOp::getCanonicalizationPatterns(patterns, context); in runOnOperation()