diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-09-01 12:51:58 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-09-16 01:06:27 +0100 |
| commit | 1533b85d198a1dd2b1ce995b6c9d69456e56eb3f (patch) | |
| tree | 9c2926e6f646d82ff72f832fcb383e88a688f66b /reference_model/src/subgraph_traverser.cc | |
| parent | 93a1628bc3dd48d9ba099de503b586a561b4751f (diff) | |
| download | reference_model-1533b85d198a1dd2b1ce995b6c9d69456e56eb3f.tar.gz | |
Implement Conv3D kernel.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ic16e918b1a2423ad563684e29ce70d9efdbf9c02
Diffstat (limited to 'reference_model/src/subgraph_traverser.cc')
| -rw-r--r-- | reference_model/src/subgraph_traverser.cc | 1 |
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: |