File tree 1 file changed +4
-2
lines changed
src/Dialect/ONNX/Transforms
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -892,15 +892,17 @@ LogicalResult ONNXGroupNormalizationCommon(
892
892
// Reshape scale/bias from [C] to [NG x C/NG x 1 x ... x 1] with numInNorm
893
893
// 1s.
894
894
biasScaleVal.emplace_back (numGroups);
895
+ axesList.emplace_back (numGroups);
895
896
// C can be a dynamic or static value, account for that here
896
897
if (C != ShapedType::kDynamic ) {
897
898
assert (C % numGroups == 0 && " expected numGroups to divide C" );
898
899
biasScaleVal.emplace_back (C / numGroups);
900
+ axesList.emplace_back (C / numGroups);
899
901
} else {
900
902
biasScaleVal.emplace_back (ShapedType::kDynamic );
903
+ axesList.emplace_back (ShapedType::kDynamic );
901
904
}
902
- axesList.emplace_back (numGroups);
903
- axesList.emplace_back (C / numGroups);
905
+
904
906
for (int64_t i = 2 ; i <= numInNorm; ++i) {
905
907
biasScaleVal.emplace_back (1 );
906
908
axesList.emplace_back (1 );
You can’t perform that action at this time.
0 commit comments