Fix NeoOnnx 7d input (#674)

Signed-off-by: Boris Zimka <boris.zimka@abbyy.com> Co-authored-by: Boris Zimka <boris.zimka@abbyy.com> Co-authored-by: Valeriy Fedyunin <valery.fedyunin@abbyy.com>
neoml-lib · Jul 5, 2022 · 3e1d078 · 3e1d078
1 parent 905cf72
commit 3e1d078
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NeoOnnx/src/GraphInput.cpp b/NeoOnnx/src/GraphInput.cpp
@@ -46,7 +46,7 @@ CPtr<const CUserTensor> CGraphInput::AddSourceLayer( CDnn& dnn ) const
 			outputShape.Last() = 1;
 		}
 	}
-	CheckNeoOnnxSupport( outputShape.Size() < BD_Count, "Tensor has too many dimensions" );
+	CheckNeoOnnxSupport( outputShape.Size() <= BD_Count, "Tensor has too many dimensions" );
 
 	CheckNeoOnnxSupport( valueInfo.type().has_tensor_type(), "Only tensors supported for graph input values" );
 	CBlobDesc outputBlobDesc(