From 1533b85d198a1dd2b1ce995b6c9d69456e56eb3f Mon Sep 17 00:00:00 2001 From: Kevin Cheng Date: Wed, 1 Sep 2021 12:51:58 -0700 Subject: Implement Conv3D kernel. Signed-off-by: Kevin Cheng Change-Id: Ic16e918b1a2423ad563684e29ce70d9efdbf9c02 --- reference_model/src/subgraph_traverser.cc | 1 + 1 file changed, 1 insertion(+) (limited to 'reference_model/src/subgraph_traverser.cc') 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: -- cgit v1.2.1