commit c76bc0c54e17a643543ae36a8f6077b803eeda89
author Derek Chow <derekjchow@google.com> Fri Oct 13 11:30:02 2023 -0700
committer Derek Chow <derekjchow@google.com> Fri Oct 13 11:32:15 2023 -0700
tree 94545bf56049c98d20c71131bfe28ca7633eec96
parent 766e5af29d6bb159697f61540be49d6663ebbe6e

Add depthwise conv test.

Change-Id: Ib8e48ce29bb45c3bfb1208f66c30ef44719e5982

tests/tflm/BUILD

diff --git a/tests/tflm/BUILD b/tests/tflm/BUILD
index c0f17f7..7b534e6 100644
--- a/tests/tflm/BUILD
+++ b/tests/tflm/BUILD
@@ -39,6 +39,23 @@
 )
 
 kelvin_test(
+    name = "depthwise_conv_test",
+    srcs = [
+        "@tflite-micro//tensorflow/lite/micro/kernels:depthwise_conv_test.cc",
+    ],
+    hw_test_size = "enormous",
+    deps = [
+        "//crt:crt_header",
+        "@tflite-micro//tensorflow/lite/c:common",
+        "@tflite-micro//tensorflow/lite/kernels/internal:tensor",
+        "@tflite-micro//tensorflow/lite/micro:micro_utils",
+        "@tflite-micro//tensorflow/lite/micro:test_helpers",
+        "@tflite-micro//tensorflow/lite/micro/kernels:kernel_runner",
+        "@tflite-micro//tensorflow/lite/micro/testing:micro_test",
+    ],
+)
+
+kelvin_test(
     name = "leaky_relu_test",
     srcs = [
         "@tflite-micro//tensorflow/lite/micro/kernels:leaky_relu_test.cc",