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
|
licenses(["notice"]) # Apache 2.0
exports_files(["LICENSE"])
load(
"//tensorflow/core:platform/default/build_config.bzl",
"tf_additional_all_protos",
"tf_proto_library",
"tf_proto_library_py",
)
tf_proto_library(
name = "tpu_embedding_config_proto",
srcs = [
"tpu_embedding_config.proto",
],
cc_api_version = 2,
visibility = ["//visibility:public"],
)
tf_proto_library(
name = "topology_proto",
srcs = [
"topology.proto",
],
cc_api_version = 2,
visibility = ["//visibility:public"],
)
tf_proto_library_py(
name = "compilation_result_proto",
srcs = [
"compilation_result.proto",
],
protodeps = tf_additional_all_protos() + [
"//tensorflow/compiler/xla:xla_data_proto",
"//tensorflow/compiler/xla/service:hlo_proto",
],
visibility = ["//visibility:public"],
)
|