experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal changes only.

Browse Source 
PiperOrigin-RevId: 352121854
Change-Id: I9e7679109d15dd4a8ec75cfc295c18dd3c1eedff
This commit is contained in:
Qiao Zhang 2021-01-15 18:40:02 -08:00 committed by TensorFlower Gardener
parent a756df5df1
commit ee541f2cf6
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/compiler/xla/pjrt/BUILD
View File

@ -8,6 +8,16 @@ package(
licenses = ["notice"], # Apache 2.0 licenses = ["notice"], # Apache 2.0
) )
package_group(
name = "friends",
includes = [
"//tensorflow:internal",
],
packages = [
"//third_party/py/jax/...",
],
)
cc_library( cc_library(
name = "worker_thread", name = "worker_thread",
srcs = ["worker_thread.cc"], srcs = ["worker_thread.cc"],
@ -274,6 +284,7 @@ cc_library(
name = "cpu_device", name = "cpu_device",
srcs = ["cpu_device.cc"], srcs = ["cpu_device.cc"],
hdrs = ["cpu_device.h"], hdrs = ["cpu_device.h"],
visibility = [":friends"],
deps = [ deps = [
":pjrt_stream_executor_client", ":pjrt_stream_executor_client",
"//tensorflow/compiler/xla:statusor", "//tensorflow/compiler/xla:statusor",