Skip to content

Commit 81df9b6

Browse files
fixing comment
1 parent 4506b27 commit 81df9b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/Dialect/ONNX/Transforms/Decompose.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1649,7 +1649,6 @@ Value decomposeIntoPhasedConvs(PatternRewriter &rewriter, Location loc,
16491649
auto stridesArrayAttr = rewriter.getI64ArrayAttr({1, 1});
16501650
Value conv;
16511651
if (needWeightsPadding) {
1652-
// Combining the 4 phased weights into single weight.
16531652
Value conv1 = getActivationAppliedToConv(
16541653
addQDQNodesForActivationIfNeeded(rewriter.create<ONNXConvOp>(loc,
16551654
convOutputType, input, addDequantizeNodeIfNeeded(weightSlices[3]),
@@ -1737,6 +1736,7 @@ Value decomposeIntoPhasedConvs(PatternRewriter &rewriter, Location loc,
17371736
: rewriter.create<ONNXConcatOp>(loc, concatOutputType,
17381737
ValueRange{conv1, conv3, conv4, conv2}, 1);
17391738
} else {
1739+
// Combining the 4 phased weights into single weight.
17401740
bool reverseOrder = (kernelShape[0] == 4);
17411741
auto combinedConvWeightsShapedType =
17421742
weightsType.get({weightsShape[0] * 4, weightsShape[1], convKernelSize,

0 commit comments

Comments
 (0)