blob: 808f6acbd182e8ed623b6a13118a02ec0504c1c8 [file] [log] [blame]
# 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",
],
)