51fe6f6acd
1. how a customized TfLiteDelegate API looks like 2. how this customized TfLiteDelegate could be plugged into TFLite benchmark and task evaluation tools. PiperOrigin-RevId: 315821839 Change-Id: Ia1c5d13bd1711a88786f36e2ef7527497a6391c7
37 lines
889 B
Python
37 lines
889 B
Python
package(
|
|
default_visibility = [
|
|
"//visibility:public",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
cc_library(
|
|
name = "simple_delegate",
|
|
srcs = [
|
|
"simple_delegate.cc",
|
|
],
|
|
hdrs = [
|
|
"simple_delegate.h",
|
|
],
|
|
deps = [
|
|
"//tensorflow/lite:kernel_api",
|
|
"//tensorflow/lite:minimal_logging",
|
|
"//tensorflow/lite/c:common",
|
|
"//tensorflow/lite/delegates:utils",
|
|
"//tensorflow/lite/kernels/internal:compatibility",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "simple_delegate_test",
|
|
srcs = ["simple_delegate_test.cc"],
|
|
deps = [
|
|
"//tensorflow/lite:framework",
|
|
"//tensorflow/lite:kernel_api",
|
|
"//tensorflow/lite/c:common",
|
|
"//tensorflow/lite/delegates/utils/dummy_delegate",
|
|
"//tensorflow/lite/kernels:builtin_ops",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|