• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/lite:build_def.bzl", "tflite_copts")
2load("//tensorflow/lite/tools/evaluation/tasks:build_def.bzl", "task_linkopts")
3
4package(
5    default_visibility = [
6        "//visibility:public",
7    ],
8    licenses = ["notice"],
9)
10
11cc_library(
12    name = "dummy_delegate",
13    srcs = [
14        "dummy_delegate.cc",
15    ],
16    hdrs = [
17        "dummy_delegate.h",
18    ],
19    deps = [
20        "//tensorflow/lite/c:common",
21        "//tensorflow/lite/delegates/utils:simple_delegate",
22    ],
23)
24
25cc_binary(
26    name = "dummy_external_delegate.so",
27    srcs = [
28        "external_delegate_adaptor.cc",
29    ],
30    linkshared = 1,
31    linkstatic = 1,
32    deps = [
33        ":dummy_delegate",
34        "//tensorflow/lite/c:common",
35        "//tensorflow/lite/tools:command_line_flags",
36        "//tensorflow/lite/tools:logging",
37    ],
38)
39
40#### The following are for using the dummy test delegate in TFLite tooling ####
41cc_library(
42    name = "dummy_delegate_provider",
43    srcs = ["dummy_delegate_provider.cc"],
44    copts = tflite_copts(),
45    deps = [
46        ":dummy_delegate",
47        "//tensorflow/lite/tools/delegates:delegate_provider_hdr",
48    ],
49    alwayslink = 1,
50)
51
52cc_binary(
53    name = "benchmark_model_plus_dummy_delegate",
54    copts = tflite_copts(),
55    linkopts = task_linkopts(),
56    deps = [
57        ":dummy_delegate_provider",
58        "//tensorflow/lite/tools/benchmark:benchmark_model_main",
59    ],
60)
61
62cc_binary(
63    name = "inference_diff_plus_dummy_delegate",
64    copts = tflite_copts(),
65    linkopts = task_linkopts(),
66    deps = [
67        ":dummy_delegate_provider",
68        "//tensorflow/lite/tools/evaluation/tasks:task_executor_main",
69        "//tensorflow/lite/tools/evaluation/tasks/inference_diff:run_eval_lib",
70    ],
71)
72
73cc_binary(
74    name = "imagenet_classification_eval_plus_dummy_delegate",
75    copts = tflite_copts(),
76    linkopts = task_linkopts(),
77    deps = [
78        ":dummy_delegate_provider",
79        "//tensorflow/lite/tools/evaluation/tasks:task_executor_main",
80        "//tensorflow/lite/tools/evaluation/tasks/imagenet_image_classification:run_eval_lib",
81    ],
82)
83
84cc_binary(
85    name = "coco_object_detection_eval_plus_dummy_delegate",
86    copts = tflite_copts(),
87    linkopts = task_linkopts(),
88    deps = [
89        ":dummy_delegate_provider",
90        "//tensorflow/lite/tools/evaluation/tasks:task_executor_main",
91        "//tensorflow/lite/tools/evaluation/tasks/coco_object_detection:run_eval_lib",
92    ],
93)
94