Skip to content

Commit 6caf973

Browse files
nit
1 parent 81df9b6 commit 6caf973

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

src/Dialect/ONNX/Transforms/Decompose.cpp

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1768,13 +1768,8 @@ Value decomposeIntoPhasedConvs(PatternRewriter &rewriter, Location loc,
17681768
}
17691769

17701770
auto combinedConvOutputType = RankedTensorType::get(
1771-
(needWeightsPadding)
1772-
? SmallVector<int64_t>(
1773-
{convOutputShape[0], convOutputShape[1] * 4,
1774-
convOutputShape[2] + 1, convOutputShape[3] + 1})
1775-
: SmallVector<int64_t>(
1776-
{convOutputShape[0], convOutputShape[1] * 4,
1777-
convOutputShape[2], convOutputShape[3]}),
1771+
SmallVector<int64_t>({convOutputShape[0], convOutputShape[1] * 4,
1772+
convOutputShape[2], convOutputShape[3]}),
17781773
convTransposeOutputType.getElementType());
17791774
conv = getActivationAppliedToConv(
17801775
addQDQNodesForActivationIfNeeded(rewriter.create<ONNXConvOp>(loc,

0 commit comments

Comments
 (0)