Fix bazel benchmark build
This commit is contained in:
parent
b9e623dd70
commit
a7af34e403
@ -110,6 +110,7 @@ cc_library(
|
||||
"//tensorflow/lite/core/api",
|
||||
"//tensorflow/lite/kernels:op_macros",
|
||||
"//tensorflow/lite/kernels/internal:compatibility",
|
||||
"//tensorflow/lite/micro/kernels:ethosu",
|
||||
"//tensorflow/lite/micro/kernels:fully_connected",
|
||||
"//tensorflow/lite/micro/kernels:micro_ops",
|
||||
"//tensorflow/lite/schema:schema_fbs",
|
||||
|
@ -41,6 +41,7 @@ cc_binary(
|
||||
"//tensorflow/lite/micro:micro_error_reporter",
|
||||
"//tensorflow/lite/micro:micro_framework",
|
||||
"//tensorflow/lite/micro:op_resolvers",
|
||||
"//tensorflow/lite/micro/kernels:ethosu",
|
||||
"//tensorflow/lite/micro/kernels:fully_connected",
|
||||
],
|
||||
)
|
||||
|
@ -95,6 +95,26 @@ cc_library(
|
||||
}),
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "ethosu",
|
||||
srcs = select({
|
||||
"//conditions:default": [
|
||||
"ethosu.cc",
|
||||
],
|
||||
}),
|
||||
hdrs = ["ethosu.h"],
|
||||
copts = micro_copts(),
|
||||
visibility = [
|
||||
# Kernel variants need to be visible to the examples and benchmarks.
|
||||
":micro",
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/lite/c:common",
|
||||
] + select({
|
||||
"//conditions:default": [],
|
||||
}),
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "micro_ops",
|
||||
srcs = [
|
||||
|
Loading…
Reference in New Issue
Block a user