-
Notifications
You must be signed in to change notification settings - Fork 7
/
BUILD.gn
115 lines (107 loc) · 2.38 KB
/
BUILD.gn
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# Copyright 2018 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import("//common-mk/mojom_bindings_generator.gni")
import("//common-mk/pkg_config.gni")
group("all") {
deps = [
":ml_service",
]
if (use.test) {
deps += [ ":ml_service_test" ]
}
if (use.fuzzer) {
deps += [
":ml_service_impl_fuzzer",
":ml_service_load_flatbuffer_fuzzer",
]
}
}
generate_mojom_bindings("ml_mojo_bindings") {
mojo_root = "${platform2_root}"
sources = [
"mojom/graph_executor.mojom",
"mojom/machine_learning_service.mojom",
"mojom/model.mojom",
"mojom/tensor.mojom",
]
}
pkg_config("ml_common_pkg_deps") {
pkg_deps = [
"libbrillo",
"libchrome-${libbase_ver}",
"libmetrics-${libbase_ver}",
"libmojo-${libbase_ver}",
]
}
static_library("ml_common") {
include_dirs = [
"${sysroot}/usr/include/tensorflow",
"${sysroot}/usr/include/tensorflow/third_party/flatbuffers",
]
libs = [ "tensorflowlite" ]
deps = [
":ml_mojo_bindings",
]
all_dependent_configs = [ ":ml_common_pkg_deps" ]
sources = [
"daemon.cc",
"graph_executor_impl.cc",
"machine_learning_service_impl.cc",
"metrics.cc",
"model_impl.cc",
"model_metadata.cc",
"request_metrics.cc",
"tensor_view.cc",
"util.cc",
]
}
executable("ml_service") {
deps = [
":ml_common",
]
sources = [
"main.cc",
]
}
if (use.test) {
executable("ml_service_test") {
configs += [ "//common-mk:test" ]
include_dirs = [
"${sysroot}/usr/include/tensorflow",
"${sysroot}/usr/include/tensorflow/third_party/flatbuffers",
]
deps = [
":ml_common",
]
sources = [
"graph_executor_impl_test.cc",
"machine_learning_service_impl_test.cc",
"model_impl_test.cc",
"test_utils.cc",
"testrunner.cc",
"util_test.cc",
]
}
}
# Fuzzer target.
if (use.fuzzer) {
executable("ml_service_impl_fuzzer") {
configs += [ "//common-mk/common_fuzzer:common_fuzzer" ]
deps = [
":ml_common",
]
sources = [
"fuzzers/ml_service_impl_fuzzer.cc",
]
}
executable("ml_service_load_flatbuffer_fuzzer") {
configs += [ "//common-mk/common_fuzzer" ]
deps = [
":ml_common",
]
sources = [
"fuzzers/ml_service_load_flatbuffer_fuzzer.cc",
]
}
}