Skip to content

Commit 94533e1

Browse files
committed
Determine layout
1 parent 0689c70 commit 94533e1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

modules/dnn/src/ie_ngraph.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -769,8 +769,14 @@ static InferenceEngine::Layout estimateLayout(const Mat& m)
769769
{
770770
if (m.dims == 4)
771771
return InferenceEngine::Layout::NCHW;
772+
else if (m.dims == 3)
773+
return InferenceEngine::Layout::CHW;
772774
else if (m.dims == 2)
773775
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;
774780
else
775781
return InferenceEngine::Layout::ANY;
776782
}

0 commit comments

Comments
 (0)