diff options
Diffstat (limited to 'tensorflow/core/platform/default/build_config.bzl')
-rw-r--r-- | tensorflow/core/platform/default/build_config.bzl | 76 |
1 files changed, 45 insertions, 31 deletions
diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl index c20db73079..f6eb40f600 100644 --- a/tensorflow/core/platform/default/build_config.bzl +++ b/tensorflow/core/platform/default/build_config.bzl @@ -1,7 +1,7 @@ # Platform-specific build configurations. -load("//google/protobuf:protobuf.bzl", "cc_proto_library") -load("//google/protobuf:protobuf.bzl", "py_proto_library") +load("@protobuf//:protobuf.bzl", "cc_proto_library") +load("@protobuf//:protobuf.bzl", "py_proto_library") # configure may change the following line to True WITH_GCP_SUPPORT = False @@ -31,29 +31,39 @@ def tf_proto_library_cc(name, srcs = [], has_services = None, cc_api_version = 2, go_api_version = 2, java_api_version = 2, py_api_version = 2): - native.filegroup(name=name + "_proto_srcs", - srcs=srcs + tf_deps(deps, "_proto_srcs"), - testonly=testonly,) + native.filegroup( + name = name + "_proto_srcs", + srcs = srcs + tf_deps(deps, "_proto_srcs"), + testonly = testonly, + ) use_grpc_plugin = None if cc_grpc_version: use_grpc_plugin = True - cc_proto_library(name=name + "_cc", - srcs=srcs + tf_deps(deps, "_proto_srcs"), - deps=deps + ["//google/protobuf:cc_wkt_protos"], - cc_libs = cc_libs + ["//google/protobuf:protobuf"], - use_grpc_plugin = use_grpc_plugin, - testonly=testonly, - visibility=visibility,) + cc_proto_library( + name = name + "_cc", + srcs = srcs + tf_deps(deps, "_proto_srcs"), + deps = deps + ["@protobuf//:cc_wkt_protos"], + cc_libs = cc_libs + ["@protobuf//:protobuf"], + protoc = "@protobuf//:protoc", + default_runtime = "@protobuf//:protobuf", + use_grpc_plugin = use_grpc_plugin, + testonly = testonly, + visibility = visibility, + ) def tf_proto_library_py(name, srcs=[], deps=[], visibility=[], testonly=0, srcs_version="PY2AND3"): - py_proto_library(name = name + "_py", - srcs = srcs, - srcs_version = srcs_version, - deps = deps, - visibility = visibility, - testonly = testonly) + py_proto_library( + name = name + "_py", + srcs = srcs, + srcs_version = srcs_version, + deps = deps, + protoc = "@protobuf//:protoc", + default_runtime = "@protobuf//:protobuf_python", + visibility = visibility, + testonly = testonly, + ) def tf_proto_library(name, srcs = [], has_services = None, deps = [], visibility = [], testonly = 0, @@ -61,19 +71,23 @@ def tf_proto_library(name, srcs = [], has_services = None, cc_api_version = 2, go_api_version = 2, java_api_version = 2, py_api_version = 2): - tf_proto_library_cc(name=name, - srcs=srcs + tf_deps(deps, "_proto_srcs"), - deps=deps, - cc_libs=cc_libs, - testonly=testonly, - visibility=visibility,) - - tf_proto_library_py(name=name, - srcs=srcs + tf_deps(deps, "_proto_srcs"), - srcs_version="PY2AND3", - deps=deps + ["//google/protobuf:protobuf_python"], - testonly=testonly, - visibility=visibility,) + tf_proto_library_cc( + name = name, + srcs = srcs + tf_deps(deps, "_proto_srcs"), + deps = deps, + cc_libs = cc_libs, + testonly = testonly, + visibility = visibility, + ) + + tf_proto_library_py( + name = name, + srcs = srcs + tf_deps(deps, "_proto_srcs"), + srcs_version = "PY2AND3", + deps = deps + ["@protobuf//:protobuf_python"], + testonly = testonly, + visibility = visibility, + ) def tf_additional_lib_srcs(): return [ |