From 812731d38d086e8d6192fbea0281957314bf552c Mon Sep 17 00:00:00 2001 From: Jens Elofsson Date: Wed, 3 Apr 2019 16:49:01 +0200 Subject: [PATCH] Add the conv kernel to the BUILD file. --- tensorflow/lite/experimental/micro/kernels/BUILD | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tensorflow/lite/experimental/micro/kernels/BUILD b/tensorflow/lite/experimental/micro/kernels/BUILD index 451eed28528..3ae760ca1ed 100644 --- a/tensorflow/lite/experimental/micro/kernels/BUILD +++ b/tensorflow/lite/experimental/micro/kernels/BUILD @@ -13,6 +13,7 @@ load( cc_library( name = "micro_ops", srcs = [ + "conv.cc", "depthwise_conv.cc", "fully_connected.cc", "softmax.cc", @@ -133,3 +134,16 @@ tflite_micro_cc_test( "//tensorflow/lite/experimental/micro/testing:micro_test", ], ) + +tflite_micro_cc_test( + name = "conv_test", + srcs = [ + "conv_test.cc", + ], + deps = [ + ":all_ops_resolver", + "//tensorflow/lite/c:c_api_internal", + "//tensorflow/lite/experimental/micro:micro_framework", + "//tensorflow/lite/experimental/micro/testing:micro_test", + ], +)