Lines Matching defs:collapseShapeOp

1938   LogicalResult matchAndRewrite(CollapseShapeOp collapseShapeOp,
1940     auto castOp = collapseShapeOp.getSrc().getDefiningOp<tensor::CastOp>();
1947         srcType, collapseShapeOp.getReassociationMaps());
1949     if (newResultType == collapseShapeOp.getResultType()) {
1950       rewriter.modifyOpInPlace(collapseShapeOp, [&]() {
1951         collapseShapeOp.getSrcMutable().assign(castOp.getSource());
1955           collapseShapeOp.getLoc(), newResultType, castOp.getSource(),
1956           collapseShapeOp.getReassociation());
1958           collapseShapeOp, collapseShapeOp.getResultType(), newOp);
2013     auto collapseShapeOp = dimOp.getSource().getDefiningOp<CollapseShapeOp>();
2014     if (!collapseShapeOp)
2020         dim.value() >= collapseShapeOp.getResultType().getRank())
2024     RankedTensorType resultType = collapseShapeOp.getResultType();
2030         collapseShapeOp.getReassociationIndices()[*dim];
2038           dimOp.getLoc(), collapseShapeOp.getSrc(), it.value()));