Refactoring some dependencies within TPU Driver
PiperOrigin-RevId: 286469337 Change-Id: I37539535a447cf929ebeb95745490fd4beb10ded
This commit is contained in:
parent
0a01852db7
commit
ae98e590af
@ -61,6 +61,7 @@ cc_library(
|
|||||||
hdrs = ["grpc_tpu_driver.h"],
|
hdrs = ["grpc_tpu_driver.h"],
|
||||||
deps = [
|
deps = [
|
||||||
":tpu_driver",
|
":tpu_driver",
|
||||||
|
"//tensorflow:grpc++",
|
||||||
"//tensorflow/core/platform:logging",
|
"//tensorflow/core/platform:logging",
|
||||||
"//tensorflow/compiler/xla:status",
|
"//tensorflow/compiler/xla:status",
|
||||||
"//tensorflow/compiler/xla:util",
|
"//tensorflow/compiler/xla:util",
|
||||||
|
@ -33,5 +33,4 @@ def external_deps():
|
|||||||
"@com_google_absl//absl/synchronization",
|
"@com_google_absl//absl/synchronization",
|
||||||
"@com_google_absl//absl/time",
|
"@com_google_absl//absl/time",
|
||||||
"@com_google_absl//absl/types:span",
|
"@com_google_absl//absl/types:span",
|
||||||
"//tensorflow:grpc++",
|
|
||||||
]
|
]
|
||||||
|
Loading…
Reference in New Issue
Block a user