Lines Matching defs:reshapeOp

822 fuseWithReshapeByExpansion(LinalgOp linalgOp, Operation *reshapeOp,
830   auto expandingReshapeOp = dyn_cast<tensor::ExpandShapeOp>(*reshapeOp);
831   auto collapsingReshapeOp = dyn_cast<tensor::CollapseShapeOp>(*reshapeOp);
980       tensor::CollapseShapeOp reshapeOp =
982       if (!reshapeOp)
992           fuseWithReshapeByExpansion(linalgOp, reshapeOp, opOperand, rewriter);
1016     tensor::CollapseShapeOp reshapeOp =
1018     if (!reshapeOp)
1020     if (!reshapeOp->hasOneUse())
1031         reshapeOp.getReassociationIndices();
1039     RankedTensorType expandedType = reshapeOp.getSrcType();
1055         loc, expandedPaddedType, reshapeOp.getSrc(), newLow, newHigh,
1079   LogicalResult matchAndRewrite(tensor::ExpandShapeOp reshapeOp,
1082     auto producerResult = dyn_cast<OpResult>(reshapeOp.getSrc());
1084       return rewriter.notifyMatchFailure(reshapeOp,
1090       return rewriter.notifyMatchFailure(reshapeOp,
1098           reshapeOp, "failed preconditions of fusion with producer generic op");
1101     if (!controlFoldingReshapes(&reshapeOp.getSrcMutable())) {
1102       return rewriter.notifyMatchFailure(reshapeOp,
1108             producer, reshapeOp,
1112       return rewriter.notifyMatchFailure(reshapeOp,
1121         (*replacementValues)[cast<OpResult>(reshapeOp.getSrc())
1127     rewriter.replaceOp(reshapeOp, reshapeReplacement);
1784       tensor::ExpandShapeOp reshapeOp =
1786       if (!reshapeOp)
1791                                            reshapeOp.getReassociationIndices());
1825     tensor::ExpandShapeOp reshapeOp =
1827     if (!reshapeOp)
1829     if (!reshapeOp->hasOneUse())
1840         reshapeOp.getReassociationIndices();
1853     RankedTensorType collapsedType = reshapeOp.getSrcType();
1857         getMixedValues(reshapeOp.getStaticOutputShape(),
1858                        reshapeOp.getOutputShape(), rewriter));
1862     Location loc = reshapeOp->getLoc();
1879         loc, collapsedPaddedType, reshapeOp.getSrc(), newLow, newHigh,