diff --git a/tensorflow/lite/micro/examples/image_recognition_experimental/BUILD b/tensorflow/lite/micro/examples/image_recognition_experimental/BUILD new file mode 100644 index 00000000000..d3bcd69d1c7 --- /dev/null +++ b/tensorflow/lite/micro/examples/image_recognition_experimental/BUILD @@ -0,0 +1,36 @@ +# Description: +# TensorFlow Lite for Microcontrollers image recognition example. + +load( + "//tensorflow/lite/micro/testing:micro_test.bzl", + "tflite_micro_cc_test", +) + +licenses(["notice"]) # Apache 2.0 + +cc_library( + name = "image_model_data", + srcs = [ + "first_10_cifar_images.cc", + "image_recognition_model.cc", + ], + hdrs = [ + "first_10_cifar_images.h", + "image_recognition_model.h", + "util.h", + ], +) + +tflite_micro_cc_test( + name = "image_recognition_test", + srcs = ["image_recognition_test.cc"], + deps = [ + ":image_model_data", + "//tensorflow/lite:schema_fbs_version", + "//tensorflow/lite/micro:micro_error_reporter", + "//tensorflow/lite/micro:micro_framework", + "//tensorflow/lite/micro:op_resolvers", + "//tensorflow/lite/micro/testing:micro_test", + "//tensorflow/lite/schema:schema_fbs", + ], +) diff --git a/tensorflow/lite/micro/examples/image_recognition_experimental/Makefile.inc b/tensorflow/lite/micro/examples/image_recognition_experimental/Makefile.inc index 2fdfb0e6779..76b21cb2580 100644 --- a/tensorflow/lite/micro/examples/image_recognition_experimental/Makefile.inc +++ b/tensorflow/lite/micro/examples/image_recognition_experimental/Makefile.inc @@ -20,6 +20,7 @@ tensorflow/lite/micro/examples/image_recognition_experimental/image_recognition_ $(MAKEFILE_DIR)/downloads/image_recognition_model/image_recognition_model.cc IMAGE_RECOGNITION_TEST_HDRS := \ +tensorflow/lite/micro/examples/image_recognition_experimental/first_10_cifar_images.h \ tensorflow/lite/micro/examples/image_recognition_experimental/image_recognition_model.h \ tensorflow/lite/micro/examples/image_recognition_experimental/util.h