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/ops/op_factory.cc | 6 ++++++ 1 file changed, 6 insertions(+) (limited to '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 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, 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); -- cgit v1.2.1