Add depthwise conv test. Change-Id: Ib8e48ce29bb45c3bfb1208f66c30ef44719e5982
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",