Update Decompose.cpp part 3

hamptonm1 · web-flow · commit 327e6efb2953 · 2024-09-11T19:47:33.000-04:00
Signed-off-by: hamptonm1 &lt;79232909+hamptonm1@users.noreply.github.com&gt;
diff --git a/src/Dialect/ONNX/Transforms/Decompose.cpp b/src/Dialect/ONNX/Transforms/Decompose.cpp
@@ -906,7 +906,7 @@ LogicalResult ONNXGroupNormalizationCommon(
       axesList.emplace_back(1);
     }
     // Reshape instead of unsqueeze (use biasScaleShape)
-    axes = create.onnx.constantInt64(axisList);
+    axes = create.onnx.constantInt64(axesList);
     biasScaleType = RankedTensorType::get(biasScaleVal, elementType);
     newScale = create.onnx.reshape(biasScaleType, scale, axes);
     newBias = create.onnx.reshape(biasScaleType, bias, axes);

Original file line number	Diff line number	Diff line change
`@@ -906,7 +906,7 @@ LogicalResult ONNXGroupNormalizationCommon(`
`906`	`906`	`axesList.emplace_back(1);`
`907`	`907`	`}`
`908`	`908`	`// Reshape instead of unsqueeze (use biasScaleShape)`
`909`		`- axes = create.onnx.constantInt64(axisList);`
	`909`	`+ axes = create.onnx.constantInt64(axesList);`
`910`	`910`	`biasScaleType = RankedTensorType::get(biasScaleVal, elementType);`
`911`	`911`	`newScale = create.onnx.reshape(biasScaleType, scale, axes);`
`912`	`912`	`newBias = create.onnx.reshape(biasScaleType, bias, axes);`