aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/subgraph_traverser.cc
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/subgraph_traverser.cc')
-rw-r--r--reference_model/src/subgraph_traverser.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc
index ef7bae6..4dba669 100644
--- a/reference_model/src/subgraph_traverser.cc
+++ b/reference_model/src/subgraph_traverser.cc
@@ -116,6 +116,7 @@ int SubgraphTraverser::initializeGraph()
switch (op->GetOp())
{
case Op_CONV2D:
+ case Op_CONV3D:
case Op_DEPTHWISE_CONV2D:
case Op_TRANSPOSE_CONV2D:
case Op_FULLY_CONNECTED: