Skip to content

Commit aaefa05

Browse files
nfiedeltensorflower-gardener
nfiedel
authored and
tensorflower-gardener
committed
Phase 1 of moving TensorFlow Serving SessionBundle to tensorflow/contrib.
Change: 123975418
1 parent b1bf065 commit aaefa05

20 files changed

Lines changed: 2915 additions & 0 deletions

‎tensorflow/BUILD‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,8 @@ filegroup(
8484
"//tensorflow/contrib/quantization:all_files",
8585
"//tensorflow/contrib/quantization/kernels:all_files",
8686
"//tensorflow/contrib/quantization/tools:all_files",
87+
"//tensorflow/contrib/session_bundle:all_files",
88+
"//tensorflow/contrib/session_bundle/example:all_files",
8789
"//tensorflow/contrib/skflow:all_files",
8890
"//tensorflow/contrib/slim:all_files",
8991
"//tensorflow/contrib/tensor_forest:all_files",

‎tensorflow/contrib/session_bundle/BUILD‎

Lines changed: 166 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,166 @@
1+
# Description: Tensorflow Serving session bundle.
2+
3+
package(
4+
default_visibility = ["//visibility:public"],
5+
features = [
6+
"-layering_check",
7+
],
8+
)
9+
10+
licenses(["notice"]) # Apache 2.0
11+
12+
exports_files(["LICENSE"])
13+
14+
filegroup(
15+
name = "all_files",
16+
srcs = glob(
17+
["**/*"],
18+
exclude = [
19+
"**/METADATA",
20+
"**/OWNERS",
21+
"g3doc/sitemap.md",
22+
],
23+
),
24+
)
25+
26+
load("//tensorflow/core:platform/default/build_config.bzl", "tf_proto_library")
27+
28+
py_library(
29+
name = "exporter",
30+
srcs = ["exporter.py"],
31+
srcs_version = "PY2AND3",
32+
deps = [
33+
":gc",
34+
":manifest_proto_py",
35+
"//tensorflow:tensorflow_py",
36+
],
37+
)
38+
39+
py_test(
40+
name = "exporter_test",
41+
size = "small",
42+
srcs = [
43+
"exporter_test.py",
44+
],
45+
srcs_version = "PY2AND3",
46+
visibility = ["//visibility:private"],
47+
deps = [
48+
":exporter",
49+
":gc",
50+
":manifest_proto_py",
51+
"//tensorflow:tensorflow_py",
52+
],
53+
)
54+
55+
py_library(
56+
name = "gc",
57+
srcs = ["gc.py"],
58+
srcs_version = "PY2AND3",
59+
deps = [
60+
"//tensorflow:tensorflow_py",
61+
],
62+
)
63+
64+
py_test(
65+
name = "gc_test",
66+
srcs = [
67+
"gc_test.py",
68+
],
69+
srcs_version = "PY2AND3",
70+
visibility = ["//visibility:private"],
71+
deps = [
72+
":gc",
73+
"//tensorflow:tensorflow_py",
74+
],
75+
)
76+
77+
cc_library(
78+
name = "session_bundle",
79+
srcs = ["session_bundle.cc"],
80+
hdrs = ["session_bundle.h"],
81+
deps = [
82+
":manifest_proto_cc",
83+
":signature",
84+
"//tensorflow/core:core_cpu",
85+
"//tensorflow/core:framework",
86+
"//tensorflow/core:lib",
87+
"//tensorflow/core:protos_all_cc",
88+
"//tensorflow/core:tensorflow_opensource",
89+
],
90+
)
91+
92+
cc_test(
93+
name = "session_bundle_test",
94+
size = "small",
95+
srcs = ["session_bundle_test.cc"],
96+
data = [
97+
"//tensorflow/contrib/session_bundle/example:half_plus_two",
98+
],
99+
# Link in all registered kernels.
100+
linkstatic = 1,
101+
visibility = ["//visibility:private"],
102+
deps = [
103+
":session_bundle",
104+
":test_util",
105+
"//tensorflow/core:core_cpu",
106+
"//tensorflow/core:framework",
107+
"//tensorflow/core:lib",
108+
"//tensorflow/core:test",
109+
"//tensorflow/core:test_main",
110+
"//tensorflow/core:testlib",
111+
],
112+
)
113+
114+
cc_library(
115+
name = "signature",
116+
srcs = ["signature.cc"],
117+
hdrs = ["signature.h"],
118+
deps = [
119+
":manifest_proto_cc",
120+
"//tensorflow/core:core_cpu",
121+
"//tensorflow/core:framework",
122+
"//tensorflow/core:lib",
123+
"//tensorflow/core:tensorflow_opensource",
124+
],
125+
)
126+
127+
cc_test(
128+
name = "signature_test",
129+
size = "small",
130+
srcs = ["signature_test.cc"],
131+
visibility = ["//visibility:private"],
132+
deps = [
133+
":manifest_proto_cc",
134+
":signature",
135+
":test_util",
136+
"//tensorflow/core:core_cpu",
137+
"//tensorflow/core:framework",
138+
"//tensorflow/core:lib",
139+
"//tensorflow/core:protos_all_cc",
140+
"//tensorflow/core:tensorflow_opensource",
141+
"//tensorflow/core:test",
142+
"//tensorflow/core:test_main",
143+
"//tensorflow/core:testlib",
144+
],
145+
)
146+
147+
cc_library(
148+
name = "test_util",
149+
testonly = 1,
150+
srcs = ["test_util.cc"],
151+
hdrs = ["test_util.h"],
152+
visibility = ["//visibility:private"],
153+
deps = [
154+
"//tensorflow/core",
155+
"//tensorflow/core:lib",
156+
"//tensorflow/core:test",
157+
],
158+
)
159+
160+
tf_proto_library(
161+
name = "manifest_proto",
162+
srcs = ["manifest.proto"],
163+
cc_api_version = 2,
164+
py_api_version = 2,
165+
visibility = ["//visibility:public"],
166+
)

0 commit comments

Comments
 (0)