diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/message_wrappers.h')
| -rw-r--r-- | tensorflow/core/distributed_runtime/message_wrappers.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/message_wrappers.h b/tensorflow/core/distributed_runtime/message_wrappers.h index 79fa6f926e..1f7cdb98a4 100644 --- a/tensorflow/core/distributed_runtime/message_wrappers.h +++ b/tensorflow/core/distributed_runtime/message_wrappers.h @@ -302,6 +302,9 @@ class MutableRunGraphRequestWrapper : public RunGraphRequestWrapper { virtual Status AddSendFromRunStepRequest( const RunStepRequestWrapper& run_step_request, size_t i, const string& send_key) = 0; + virtual Status AddSendFromRunCallableRequest( + const RunCallableRequest& run_callable_request, size_t i, + const string& send_key) = 0; virtual void add_recv_key(const string& recv_key) = 0; virtual void set_is_partial(bool is_partial) = 0; @@ -334,6 +337,9 @@ class InMemoryRunGraphRequest : public MutableRunGraphRequestWrapper { Status AddSendFromRunStepRequest( const RunStepRequestWrapper& run_step_request, size_t i, const string& send_key) override; + Status AddSendFromRunCallableRequest( + const RunCallableRequest& run_callable_request, size_t i, + const string& send_key) override; void add_recv_key(const string& recv_key) override; void set_is_partial(bool is_partial) override; void set_is_last_partial_run(bool is_last_partial_run) override; @@ -385,6 +391,9 @@ class MutableProtoRunGraphRequest : public MutableRunGraphRequestWrapper { Status AddSendFromRunStepRequest( const RunStepRequestWrapper& run_step_request, size_t i, const string& send_key) override; + Status AddSendFromRunCallableRequest( + const RunCallableRequest& run_callable_request, size_t i, + const string& send_key) override; void add_recv_key(const string& recv_key) override; void set_is_partial(bool is_partial) override; void set_is_last_partial_run(bool is_last_partial_run) override; |