diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2017-11-01 15:17:51 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-01 15:21:32 -0700 |
| commit | 2ce0b9149741105795083c4bae8fb0b85cb9659d (patch) | |
| tree | 9a1ed8b1160b3f2aea043dc8b9c01715d947197f /tensorflow/core/distributed_runtime/rpc/BUILD | |
| parent | 3540a4277466d714a975ec4a69d9294b4f65438c (diff) | |
Factor out RPCState into its own header.
PiperOrigin-RevId: 174243882
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/BUILD')
| -rw-r--r-- | tensorflow/core/distributed_runtime/rpc/BUILD | 25 |
1 files changed, 22 insertions, 3 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/BUILD b/tensorflow/core/distributed_runtime/rpc/BUILD index 5190288e88..51e499d3f5 100644 --- a/tensorflow/core/distributed_runtime/rpc/BUILD +++ b/tensorflow/core/distributed_runtime/rpc/BUILD @@ -53,9 +53,11 @@ cc_library( srcs = ["grpc_util.cc"], hdrs = ["grpc_util.h"], deps = [ - "//tensorflow/core:lib", - "@grpc//:grpc++_unsecure", "@grpc//:grpc_unsecure", + "@grpc//:grpc++_unsecure", + "//tensorflow/core:lib", + # Required to be able to overload TensorResponse parsing. + "//tensorflow/core/distributed_runtime:tensor_coding", ], ) @@ -71,17 +73,33 @@ cc_library( ) cc_library( + name = "grpc_state", + srcs = [], + hdrs = ["grpc_state.h"], + deps = [ + ":grpc_client_cq_tag", + ":grpc_util", + "//tensorflow/core:lib", + "//tensorflow/core/distributed_runtime:call_options", + "//tensorflow/core/distributed_runtime:tensor_coding", + "@grpc//:grpc++_unsecure", + ], +) + +cc_library( name = "grpc_remote_worker", srcs = ["grpc_remote_worker.cc"], hdrs = ["grpc_remote_worker.h"], deps = [ ":grpc_client_cq_tag", + ":grpc_state", ":grpc_util", ":grpc_worker_service_impl", "//tensorflow/core:core_cpu_internal", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", "//tensorflow/core:worker_proto_cc", + "//tensorflow/core/distributed_runtime:tensor_coding", "//tensorflow/core/distributed_runtime:worker_cache_logger", "//tensorflow/core/distributed_runtime:worker_interface", "@grpc//:grpc++_unsecure", @@ -185,7 +203,7 @@ cc_library( ":grpc_namespace_compat", ":grpc_serialization_traits", "//tensorflow/core:worker_proto_cc", - "//tensorflow/core/distributed_runtime:worker_interface", + "//tensorflow/core/distributed_runtime:tensor_coding", "@grpc//:grpc++_unsecure", ], ) @@ -263,6 +281,7 @@ cc_library( "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core/distributed_runtime:base_rendezvous_mgr", + "//tensorflow/core/distributed_runtime:tensor_coding", "//tensorflow/core/distributed_runtime:worker_cache", "//tensorflow/core/distributed_runtime:worker_env", "//tensorflow/core/distributed_runtime:worker_interface", |