commit 1533b85d198a1dd2b1ce995b6c9d69456e56eb3f
author Kevin Cheng <kevin.cheng@arm.com> Wed Sep 01 12:51:58 2021 -0700
committer Kevin Cheng <kevin.cheng@arm.com> Thu Sep 16 01:06:27 2021 +0100
tree 9c2926e6f646d82ff72f832fcb383e88a688f66b
parent 93a1628bc3dd48d9ba099de503b586a561b4751f

Implement Conv3D kernel.

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ic16e918b1a2423ad563684e29ce70d9efdbf9c02

reference_model/src/ops/op_factory.cc

diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc
index 193b2af..3bc55a8 100644
--- a/reference_model/src/ops/op_factory.cc
+++ b/reference_model/src/ops/op_factory.cc
@@ -64,6 +64,12 @@
             DEF_FACTORY_TWO_TYPE(OpConv2d, INT8, INT8);
             DEF_FACTORY_TWO_TYPE(OpConv2d, INT16, INT8);
             break;
+        case Op_CONV3D:
+            DEF_FACTORY_TWO_TYPE(OpConv3d, FLOAT, FLOAT);
+            DEF_FACTORY_TWO_TYPE(OpConv3d, INT8, INT4);
+            DEF_FACTORY_TWO_TYPE(OpConv3d, INT8, INT8);
+            DEF_FACTORY_TWO_TYPE(OpConv3d, INT16, INT8);
+            break;
         case Op_DEPTHWISE_CONV2D:
             DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, FLOAT, FLOAT);
             DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, INT8, INT4);