Skip to content

Commit 936c529

Browse files
making sure we have batch=1
1 parent d893b73 commit 936c529

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/Dialect/ONNX/Transforms/Decompose.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -851,6 +851,10 @@ bool ShouldDecomposeConvTransposeOpToPhasedConvs(Value convTransposeResult,
851851
bool fourPhaseDecomposition = (stridesShape[0] == 2);
852852
bool ninePhaseDecomposition = (stridesShape[0] == 3);
853853
if (fourPhaseDecomposition) {
854+
if (outputShape[0] != 1) {
855+
// Currently support batch=1
856+
return false;
857+
}
854858
if (kernelShape[0] == 6 && padsShape[0] == 2 &&
855859
llvm::all_equal(padsShape)) {
856860
// Currently support only with pads [2, 2, 2, 2]

0 commit comments

Comments
 (0)