Skip to content

Commit 200e9bc

Browse files
committed
Use helper function to create TOSA const shape
1 parent ffc3648 commit 200e9bc

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

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)