33 lines
821 B
Python
33 lines
821 B
Python
load("//tensorflow/lite:build_def.bzl", "tflite_copts")
|
|
load("//tensorflow/lite/tools/evaluation/tasks:build_def.bzl", "task_linkopts")
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//visibility:public",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
cc_library(
|
|
name = "task_executor",
|
|
hdrs = ["task_executor.h"],
|
|
copts = tflite_copts(),
|
|
linkopts = task_linkopts(),
|
|
deps = [
|
|
"//tensorflow/lite/tools/evaluation/proto:evaluation_config_cc_proto",
|
|
"@com_google_absl//absl/types:optional",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "task_executor_main",
|
|
srcs = ["task_executor_main.cc"],
|
|
copts = tflite_copts(),
|
|
linkopts = task_linkopts(),
|
|
deps = [
|
|
":task_executor",
|
|
"//tensorflow/lite/tools:logging",
|
|
"@com_google_absl//absl/types:optional",
|
|
],
|
|
)
|