fix

Signed-off-by: chentong319 <[email protected]>
chentong319 · Feb 21, 2024 · 7f02511 · 7f02511
1 parent 8f64328
commit 7f02511
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Transform/ONNX/Decompose.cpp b/src/Transform/ONNX/Decompose.cpp
@@ -905,7 +905,8 @@ struct GroupNormIntoLayerNormPattern
     Type inputShapeType =
         RankedTensorType::get({inputRank}, rewriter.getI64Type());
     Value inputShape = create.onnx.shape(inputShapeType, input);
-    Value Y = create.onnx.reshape(inputType, layerNormY, inputShape);
+    Type outputType = groupNormOp.getY().getType();
+    Value Y = create.onnx.reshape(outputType, layerNormY, inputShape);
     // Replace operation.
     rewriter.replaceOp(groupNormOp, Y);
     return success();