diff options
Diffstat (limited to 'tensorflow/contrib/verbs/BUILD')
-rw-r--r-- | tensorflow/contrib/verbs/BUILD | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/tensorflow/contrib/verbs/BUILD b/tensorflow/contrib/verbs/BUILD index e747fa4c9e..5f062cde89 100644 --- a/tensorflow/contrib/verbs/BUILD +++ b/tensorflow/contrib/verbs/BUILD @@ -66,9 +66,8 @@ cc_library( ":grpc_verbs_service_impl", ":rdma_mgr", ":verbs_service_proto_cc", - "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", "//tensorflow/core/distributed_runtime:session_mgr", - "//tensorflow/core/distributed_runtime:worker_env", "//tensorflow/core/distributed_runtime/rpc:async_service_interface", "//tensorflow/core/distributed_runtime/rpc:grpc_call", "//tensorflow/core/distributed_runtime/rpc:grpc_util", @@ -107,8 +106,8 @@ cc_library( hdrs = ["rdma_rendezvous_mgr.h"], deps = [ ":rdma_mgr", + ":verbs_util", "//tensorflow/core:core_cpu_internal", - "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core/distributed_runtime:base_rendezvous_mgr", "//tensorflow/core/distributed_runtime:worker_env", @@ -122,10 +121,11 @@ cc_library( deps = [ ":grpc_verbs_client", ":rdma", + ":verbs_service_proto_cc", "//tensorflow/core:core_cpu_internal", - "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", + "//tensorflow/core/distributed_runtime:session_mgr", "//tensorflow/core/distributed_runtime:worker_env", "//tensorflow/core/distributed_runtime/rpc:grpc_channel", "//tensorflow/core/distributed_runtime/rpc:grpc_worker_cache", @@ -144,6 +144,7 @@ cc_library( ":verbs_util", "//tensorflow/core:core_cpu_internal", "//tensorflow/core:framework", + "//tensorflow/core:framework_internal", "//tensorflow/core:gpu_runtime", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", @@ -162,6 +163,8 @@ cc_library( ":grpc_verbs_service", ":rdma_mgr", ":rdma_rendezvous_mgr", + "//tensorflow/core:lib", + "//tensorflow/core/distributed_runtime:server_lib", "//tensorflow/core/distributed_runtime/rpc:grpc_server_lib", ], alwayslink = 1, |