Skip to content

Commit ba471a1

Browse files
authored
Still at it
Signed-off-by: hamptonm1 <[email protected]>
1 parent 327e6ef commit ba471a1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/Dialect/ONNX/Transforms/Decompose.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -892,15 +892,17 @@ LogicalResult ONNXGroupNormalizationCommon(
892892
// Reshape scale/bias from [C] to [NG x C/NG x 1 x ... x 1] with numInNorm
893893
// 1s.
894894
biasScaleVal.emplace_back(numGroups);
895+
axesList.emplace_back(numGroups);
895896
// C can be a dynamic or static value, account for that here
896897
if (C != ShapedType::kDynamic) {
897898
assert(C % numGroups == 0 && "expected numGroups to divide C");
898899
biasScaleVal.emplace_back(C / numGroups);
900+
axesList.emplace_back(C / numGroups);
899901
} else {
900902
biasScaleVal.emplace_back(ShapedType::kDynamic);
903+
axesList.emplace_back(ShapedType::kDynamic);
901904
}
902-
axesList.emplace_back(numGroups);
903-
axesList.emplace_back(C / numGroups);
905+
904906
for (int64_t i = 2; i <= numInNorm; ++i) {
905907
biasScaleVal.emplace_back(1);
906908
axesList.emplace_back(1);

0 commit comments

Comments
 (0)