• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1licenses(["restricted"])
2
3package(default_visibility = ["//visibility:public"])
4
5load(
6    "//tensorflow/core:platform/default/build_config_root.bzl",
7    "if_static",
8)
9
10cc_library(
11    name = "interpreter_transfer_manager",
12    srcs = ["interpreter_transfer_manager.cc"],
13    hdrs = ["interpreter_transfer_manager.h"],
14    deps = [
15        "//tensorflow/compiler/xla:util",
16        "//tensorflow/compiler/xla:xla_data_proto",
17        "//tensorflow/compiler/xla/service:generic_transfer_manager",
18        "//tensorflow/compiler/xla/service:transfer_manager",
19        "//tensorflow/compiler/xla/service/interpreter:platform_id",
20        "//tensorflow/core:lib",
21        "//tensorflow/core:stream_executor_no_cuda",
22    ],
23    alwayslink = True,  # Contains per-platform transfer manager registration
24)
25
26cc_library(
27    name = "compiler",
28    srcs = ["compiler.cc"],
29    hdrs = ["compiler.h"],
30    deps = [
31        ":executable",
32        ":platform_id",
33        "//tensorflow/compiler/xla:status",
34        "//tensorflow/compiler/xla:status_macros",
35        "//tensorflow/compiler/xla:statusor",
36        "//tensorflow/compiler/xla:util",
37        "//tensorflow/compiler/xla:xla_data_proto",
38        "//tensorflow/compiler/xla/service:algebraic_simplifier",
39        "//tensorflow/compiler/xla/service:compiler",
40        "//tensorflow/compiler/xla/service:computation_placer",
41        "//tensorflow/compiler/xla/service:executable",
42        "//tensorflow/compiler/xla/service:flatten_call_graph",
43        "//tensorflow/compiler/xla/service:hlo",
44        "//tensorflow/compiler/xla/service:hlo_constant_folding",
45        "//tensorflow/compiler/xla/service:hlo_cost_analysis",
46        "//tensorflow/compiler/xla/service:hlo_cse",
47        "//tensorflow/compiler/xla/service:hlo_dce",
48        "//tensorflow/compiler/xla/service:hlo_module_config",
49        "//tensorflow/compiler/xla/service:hlo_pass",
50        "//tensorflow/compiler/xla/service:hlo_pass_pipeline",
51        "//tensorflow/compiler/xla/service:hlo_subcomputation_unification",
52        "//tensorflow/compiler/xla/service:inliner",
53        "//tensorflow/compiler/xla/service:layout_assignment",
54        "//tensorflow/compiler/xla/service:reshape_mover",
55        "//tensorflow/compiler/xla/service:while_loop_simplifier",
56        "//tensorflow/core:lib",
57        "//tensorflow/stream_executor",
58    ],
59    alwayslink = True,  # Contains compiler registration
60)
61
62cc_library(
63    name = "platform_id",
64    srcs = ["platform_id.cc"],
65    hdrs = ["platform_id.h"],
66    deps = [
67        "@nsync//:nsync_headers",
68        "//tensorflow/core:stream_executor_headers_lib",
69    ] + if_static(
70        ["@protobuf_archive//:protobuf"],
71        ["@protobuf_archive//:protobuf_headers"],
72    ),
73)
74
75cc_library(
76    name = "executable",
77    srcs = ["executable.cc"],
78    hdrs = ["executable.h"],
79    deps = [
80        ":executor",
81        "//tensorflow/compiler/xla:literal_util",
82        "//tensorflow/compiler/xla:shape_util",
83        "//tensorflow/compiler/xla:status_macros",
84        "//tensorflow/compiler/xla:statusor",
85        "//tensorflow/compiler/xla:types",
86        "//tensorflow/compiler/xla:util",
87        "//tensorflow/compiler/xla:xla_data_proto",
88        "//tensorflow/compiler/xla/service:executable",
89        "//tensorflow/compiler/xla/service:hlo",
90        "//tensorflow/compiler/xla/service:hlo_cost_analysis",
91        "//tensorflow/compiler/xla/service:hlo_evaluator",
92        "//tensorflow/compiler/xla/service:hlo_execution_profile",
93        "//tensorflow/compiler/xla/service:hlo_module_config",
94        "//tensorflow/compiler/xla/service:shaped_buffer",
95        "//tensorflow/compiler/xla/service:transfer_manager",
96        "//tensorflow/core:lib",
97        "//tensorflow/core:stream_executor_no_cuda",
98    ],
99)
100
101cc_library(
102    name = "platform",
103    srcs = ["platform.cc"],
104    hdrs = ["platform.h"],
105    deps = [
106        ":executor",
107        ":platform_id",
108        "//tensorflow/core:stream_executor_headers_lib",
109    ],
110    alwayslink = True,  # Registers itself with the MultiPlatformManager.
111)
112
113cc_library(
114    name = "executor",
115    srcs = ["executor.cc"],
116    hdrs = ["executor.h"],
117    deps = [
118        "//tensorflow/compiler/xla:shape_util",
119        "//tensorflow/compiler/xla:status_macros",
120        "//tensorflow/compiler/xla:xla_data_proto",
121        "//tensorflow/core:lib",
122        "//tensorflow/core:stream_executor_headers_lib",
123        "//tensorflow/core:stream_executor_no_cuda",
124    ],
125)
126
127filegroup(
128    name = "all_files",
129    srcs = glob(
130        ["**/*"],
131        exclude = [
132            "**/METADATA",
133            "**/OWNERS",
134        ],
135    ),
136)
137