| # Description: |
| # TensorFlow Lite microcontroller benchmarks. |
| package( |
| # Disabling layering_check because of http://b/177257332 |
| features = ["-layering_check"], |
| licenses = ["notice"], |
| ) |
| |
| package_group( |
| name = "micro_top_level", |
| packages = ["//tensorflow/lite/micro"], |
| ) |
| |
| cc_library( |
| name = "micro_benchmark", |
| hdrs = [ |
| "micro_benchmark.h", |
| ], |
| visibility = [ |
| "//visibility:public", |
| ], |
| deps = [ |
| "//tensorflow/lite/micro:micro_framework", |
| "//tensorflow/lite/micro:micro_log", |
| "//tensorflow/lite/micro:micro_resource_variable", |
| "//tensorflow/lite/micro:micro_time", |
| "//tensorflow/lite/micro:op_resolvers", |
| "//tensorflow/lite/micro:recording_allocators", |
| ], |
| ) |
| |
| cc_library( |
| name = "keyword_scrambled_model_data", |
| srcs = [ |
| "//tensorflow/lite/micro/models:generated_keyword_scrambled_model_cc", |
| ], |
| hdrs = [ |
| "//tensorflow/lite/micro/models:generated_keyword_scrambled_model_hdr", |
| ], |
| visibility = [ |
| ":micro_top_level", |
| ], |
| ) |
| |
| cc_binary( |
| name = "keyword_benchmark", |
| srcs = ["keyword_benchmark.cc"], |
| deps = [ |
| ":keyword_scrambled_model_data", |
| ":micro_benchmark", |
| "//tensorflow/lite/c:common", |
| "//tensorflow/lite/micro:micro_framework", |
| "//tensorflow/lite/micro:micro_log", |
| "//tensorflow/lite/micro:micro_profiler", |
| "//tensorflow/lite/micro:op_resolvers", |
| "//tensorflow/lite/micro:system_setup", |
| ], |
| ) |
| |
| cc_library( |
| name = "keyword_scrambled_8bit_model_data", |
| srcs = [ |
| "//tensorflow/lite/micro/models:generated_keyword_scrambled_8bit_model_cc", |
| ], |
| hdrs = [ |
| "//tensorflow/lite/micro/models:generated_keyword_scrambled_8bit_model_hdr", |
| ], |
| visibility = [ |
| ":micro_top_level", |
| ], |
| ) |
| |
| cc_binary( |
| name = "keyword_benchmark_8bit", |
| srcs = ["keyword_benchmark_8bit.cc"], |
| deps = [ |
| ":keyword_scrambled_8bit_model_data", |
| ":micro_benchmark", |
| "//tensorflow/lite/c:common", |
| "//tensorflow/lite/micro:micro_framework", |
| "//tensorflow/lite/micro:micro_log", |
| "//tensorflow/lite/micro:op_resolvers", |
| "//tensorflow/lite/micro:system_setup", |
| ], |
| ) |
| |
| cc_binary( |
| name = "person_detection_benchmark", |
| srcs = ["person_detection_benchmark.cc"], |
| deps = [ |
| ":micro_benchmark", |
| "//tensorflow/lite/c:common", |
| "//tensorflow/lite/micro:micro_framework", |
| "//tensorflow/lite/micro:micro_log", |
| "//tensorflow/lite/micro:micro_utils", |
| "//tensorflow/lite/micro:op_resolvers", |
| "//tensorflow/lite/micro:system_setup", |
| "//tensorflow/lite/micro/examples/person_detection:model_settings", |
| "//tensorflow/lite/micro/examples/person_detection:person_detect_model_data", |
| "//tensorflow/lite/micro/examples/person_detection:simple_images_test_data", |
| "//tensorflow/lite/schema:schema_fbs", |
| ], |
| ) |