We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0689c70 commit 94533e1Copy full SHA for 94533e1
modules/dnn/src/ie_ngraph.cpp
@@ -769,8 +769,14 @@ static InferenceEngine::Layout estimateLayout(const Mat& m)
769
{
770
if (m.dims == 4)
771
return InferenceEngine::Layout::NCHW;
772
+ else if (m.dims == 3)
773
+ return InferenceEngine::Layout::CHW;
774
else if (m.dims == 2)
775
return InferenceEngine::Layout::NC;
776
+ else if (m.dims == 1)
777
+ return InferenceEngine::Layout::C;
778
+ else if (m.dims == 5)
779
+ return InferenceEngine::Layout::NCDHW;
780
else
781
return InferenceEngine::Layout::ANY;
782
}
0 commit comments