diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-06-26 15:36:59 -0700 |
|---|---|---|
| committer |  Gunhan Gulsoy <gunan@google.com> | 2018-06-28 21:37:43 -0700 |
| commit | 24c1634196129d60568044437b6db225f6f7d721 (patch) | |
| tree | a25a9f403f400891b6cc393fba1fa43a934266f5 /tensorflow/core/distributed_runtime/eager/eager_service_impl.h | |
| parent | 528fe6ab556655985e601fba65162a134bfcd19b (diff) | |
Support shapes for remote eager tensor handles.
Since we respond with the shape, all RPCs will happen sync (note
that we may still hide the python overhead, since the op is still scheduled for
execution via the eager executor).
PiperOrigin-RevId: 202207324
Diffstat (limited to 'tensorflow/core/distributed_runtime/eager/eager_service_impl.h')
| -rw-r--r-- | tensorflow/core/distributed_runtime/eager/eager_service_impl.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/distributed_runtime/eager/eager_service_impl.h b/tensorflow/core/distributed_runtime/eager/eager_service_impl.h index ebd5269a57..b0e4aa84b9 100644 --- a/tensorflow/core/distributed_runtime/eager/eager_service_impl.h +++ b/tensorflow/core/distributed_runtime/eager/eager_service_impl.h @@ -135,7 +135,8 @@ class EagerServiceImpl { tensorflow::Status GetServerContext(uint64, ServerContext**); private: - Status ExecuteOp(const Operation& operation, ServerContext* server_context); + Status ExecuteOp(const Operation& operation, ServerContext* server_context, + QueueResponse* queue_response); const WorkerEnv* const env_; // Not owned. mutex contexts_mu_; |