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
|
"""Wrapper around cc_proto_library used inside the XLA codebase."""
load("//tensorflow/core:platform/default/build_config.bzl",
"cc_proto_library")
load("//tensorflow/core:platform/default/build_config_root.bzl",
"if_static")
# xla_proto_library() is a convenience wrapper around cc_proto_library.
def xla_proto_library(name, srcs=[], deps=[], visibility=None, testonly=0, **kwargs):
if kwargs.get('use_grpc_plugin'):
kwargs['use_grpc_namespace'] = True
cc_proto_library(name=name,
srcs=srcs,
deps=deps,
cc_libs = if_static(
["@protobuf_archive//:protobuf"],
otherwise=["@protobuf_archive//:protobuf_headers"],
),
protoc="@protobuf_archive//:protoc",
testonly=testonly,
visibility=visibility,
**kwargs)
def xla_py_grpc_library(**kwargs):
# Note: we don't currently define any special targets for Python GRPC in OSS.
_ignore = kwargs
pass
ORC_JIT_MEMORY_MAPPER_TARGETS = []
|
