forked from tensorflow/tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBUILD
More file actions
95 lines (85 loc) · 2.89 KB
/
Copy pathBUILD
File metadata and controls
95 lines (85 loc) · 2.89 KB
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
# Tensorflow protobuf utility package
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable")
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "tf_py_test")
# buildifier: disable=same-origin-load
load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library") # @unused
visibility = [
"//engedu/ml/tf_from_scratch:__pkg__",
"//third_party/cloud_tpu/convergence_tools:__subpackages__",
"//third_party/mlperf:__subpackages__",
"//tensorflow:internal",
"//tensorflow/lite/toco/python:__pkg__",
"//tensorflow_models:__subpackages__",
"//tensorflow_model_optimization:__subpackages__",
"//third_party/py/cleverhans:__subpackages__",
"//third_party/py/launchpad:__subpackages__",
"//third_party/py/reverb:__subpackages__",
"//third_party/py/neural_structured_learning:__subpackages__",
"//third_party/py/tensorflow_examples:__subpackages__",
"//third_party/py/tf_agents:__subpackages__", # For benchmarks.
"//third_party/py/tf_slim:__subpackages__",
"//third_party/py/tensorflow_docs:__subpackages__",
"//third_party/py/keras:__subpackages__",
]
package(
default_visibility = visibility,
licenses = ["notice"],
)
tf_proto_library(
name = "compare_test_proto",
testonly = 1,
srcs = ["compare_test.proto"],
cc_api_version = 2,
)
tf_py_test(
name = "protobuf_compare_test",
size = "small",
srcs = ["compare_test.py"],
main = "compare_test.py",
python_version = "PY3",
tags = ["no_pip"], # compare_test_pb2 proto is not available in pip.
deps = [
":compare_test_proto_py",
"@six_archive//:six",
"//tensorflow/python:platform_test",
#"//tensorflow/python/util",
],
)
filegroup(
name = "compare_test_proto_src",
srcs = ["compare_test.proto"],
)
# copybara:uncomment_begin(google-only)
# py_proto_library(
# name = "compare_test_py_pb2",
# testonly = 1,
# has_services = 0,
# api_version = 2,
# deps = [":compare_test_proto"],
# )
# copybara:uncomment_end
py_library(
name = "protobuf",
srcs = glob(
["*.py"],
exclude = ["*_test.py"],
),
compatible_with = get_compatible_with_portable(),
srcs_version = "PY3",
visibility = visibility + [
"//tensorflow:__pkg__",
"//third_party/py/tensorflow_core:__subpackages__",
"//third_party/py/tf_agents:__subpackages__",
"//third_party/py/tfx:__subpackages__",
],
deps = [
# global_test_configuration is added here because all major tests depend on this
# library. It isn't possible to add these test dependencies via tensorflow.bzl's
# py_test because not all tensorflow tests use tensorflow.bzl's py_test.
"//tensorflow/python:global_test_configuration",
"@com_google_protobuf//:protobuf_python",
"@six_archive//:six",
],
)