File size: 2,354 Bytes
97b6013 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
package(
default_visibility = ["//visibility:public"],
)
licenses(["notice"])
cc_library(
name = "mobile_ssd_client",
srcs = ["mobile_ssd_client.cc"],
hdrs = ["mobile_ssd_client.h"],
deps = [
"//protos:box_encodings_cc_proto",
"//protos:detections_cc_proto",
"//protos:labelmap_cc_proto",
"//protos:mobile_ssd_client_options_cc_proto",
"//utils:conversion_utils",
"//utils:ssd_utils",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/types:span",
"@com_google_glog//:glog",
"@gemmlowp",
],
)
config_setting(
name = "enable_edgetpu",
define_values = {"enable_edgetpu": "true"},
visibility = ["//visibility:public"],
)
cc_library(
name = "mobile_ssd_tflite_client",
srcs = ["mobile_ssd_tflite_client.cc"],
hdrs = ["mobile_ssd_tflite_client.h"],
defines = select({
"//conditions:default": [],
"enable_edgetpu": ["ENABLE_EDGETPU"],
}),
deps = [
":mobile_ssd_client",
"@com_google_glog//:glog",
"@com_google_absl//absl/memory",
"@org_tensorflow//tensorflow/lite:arena_planner",
"@org_tensorflow//tensorflow/lite:framework",
"@org_tensorflow//tensorflow/lite/delegates/nnapi:nnapi_delegate",
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
"//protos:anchor_generation_options_cc_proto",
"//utils:file_utils",
"//utils:ssd_utils",
] + select({
"//conditions:default": [],
"enable_edgetpu": [
"@libedgetpu//libedgetpu:header",
],
}),
alwayslink = 1,
)
cc_library(
name = "mobile_lstd_tflite_client",
srcs = ["mobile_lstd_tflite_client.cc"],
hdrs = ["mobile_lstd_tflite_client.h"],
defines = select({
"//conditions:default": [],
"enable_edgetpu": ["ENABLE_EDGETPU"],
}),
deps = [
":mobile_ssd_client",
":mobile_ssd_tflite_client",
"@com_google_glog//:glog",
"@com_google_absl//absl/base:core_headers",
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
] + select({
"//conditions:default": [],
"enable_edgetpu": [
"@libedgetpu//libedgetpu:header",
],
}),
alwayslink = 1,
)
|