Skip to content

Commit 1f31053

Browse files
committed
Use helper function to create TOSA const shape
1 parent 1861467 commit 1f31053

1 file changed

Lines changed: 1 addition & 10 deletions

File tree

stablehlo/conversions/tosa/transforms/StablehloLegalizeToTosa.cpp

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -505,16 +505,7 @@ struct ConvertStablehloReshapeOp
505505
auto resultShape = resultType.getShape();
506506
SmallVector<int64_t, 8> dimensions(resultShape.begin(), resultShape.end());
507507

508-
RankedTensorType shapeTensorType = RankedTensorType::get(
509-
{static_cast<int64_t>(dimensions.size())}, rewriter.getIndexType());
510-
511-
auto denseAttr = DenseIntElementsAttr::get(shapeTensorType, dimensions);
512-
auto shapeType =
513-
tosa::shapeType::get(rewriter.getContext(), dimensions.size());
514-
515-
auto constShapeOp =
516-
tosa::ConstShapeOp::create(rewriter, op.getLoc(), shapeType, denseAttr);
517-
508+
auto constShapeOp = getTosaConstShape(rewriter, op.getLoc(), dimensions);
518509
auto reshapeOp = tosa::ReshapeOp::create(rewriter, op.getLoc(), resultType,
519510
op.getOperand(), constShapeOp);
520511

0 commit comments

Comments
 (0)