Lines Matching refs:reshapeOp
684 fuseWithReshapeByExpansion(GenericOp genericOp, Operation *reshapeOp, in fuseWithReshapeByExpansion() argument
690 auto expandingReshapeOp = dyn_cast<tensor::ExpandShapeOp>(*reshapeOp); in fuseWithReshapeByExpansion()
691 auto collapsingReshapeOp = dyn_cast<tensor::CollapseShapeOp>(*reshapeOp); in fuseWithReshapeByExpansion()
829 tensor::CollapseShapeOp reshapeOp = in matchAndRewrite() local
831 if (!reshapeOp) in matchAndRewrite()
837 (!controlFoldingReshapes(reshapeOp->getResult(0), *opOperand))) in matchAndRewrite()
841 fuseWithReshapeByExpansion(genericOp, reshapeOp, opOperand, rewriter); in matchAndRewrite()
865 LogicalResult matchAndRewrite(tensor::ExpandShapeOp reshapeOp, in matchAndRewrite()
868 GenericOp producer = reshapeOp.getSrc().getDefiningOp<GenericOp>(); in matchAndRewrite()
873 reshapeOp->getOpOperand(0))) in matchAndRewrite()
876 producer, reshapeOp, producer.getOutputOperand(0), rewriter); in matchAndRewrite()
879 rewriter.replaceOp(reshapeOp, *replacementValues); in matchAndRewrite()
1286 auto reshapeOp = builder.create<tensor::CollapseShapeOp>( in getCollapsedOpOperand() local
1288 return reshapeOp.getResult(); in getCollapsedOpOperand()
1449 tensor::ExpandShapeOp reshapeOp = in matchAndRewrite() local
1451 if (!reshapeOp) in matchAndRewrite()
1456 reshapeOp.getReassociationIndices()); in matchAndRewrite()
1458 !controlFoldingReshapes(reshapeOp->getResult(0), *opOperand)) { in matchAndRewrite()