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.
2 parents a0008de + 94533e1 commit 2a808aeCopy full SHA for 2a808ae
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