Lines Matching defs:reshapeOp
1879 LogicalResult matchAndRewrite(TensorReshapeOp reshapeOp,
1882 if (!matchPattern(reshapeOp.getSrc(), m_Constant(&attr)))
1887 reshapeOp.getResultType(), attr.getRawData());
1888 rewriter.replaceOpWithNewOp<arith::ConstantOp>(reshapeOp, newAttr);
1899 LogicalResult matchAndRewrite(TensorReshapeOp reshapeOp,
1901 auto splatOp = reshapeOp.getSrc().template getDefiningOp<tensor::SplatOp>();
1906 reshapeOp, reshapeOp.getResultType(), splatOp.getInput());
1916 LogicalResult matchAndRewrite(TensorReshapeOp reshapeOp,
1919 reshapeOp.getSrc().template getDefiningOp<FromElementsOp>();
1923 auto shapedTy = llvm::cast<ShapedType>(reshapeOp.getType());
1928 rewriter.replaceOpWithNewOp<FromElementsOp>(reshapeOp, reshapeOp.getType(),