1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/call_options.cc b/tensorflow/core/distributed_runtime/call_options.cc
index b9d583b754..a99cec9205 100644
--- a/tensorflow/core/distributed_runtime/call_options.cc
+++ b/tensorflow/core/distributed_runtime/call_options.cc
@@ -41,4 +41,14 @@ void CallOptions::ClearCancelCallback() {
   cancel_func_ = nullptr;
 }
 
+int64 CallOptions::GetTimeout() {
+  mutex_lock l(mu_);
+  return timeout_in_ms_;
+}
+
+void CallOptions::SetTimeout(int64 ms) {
+  mutex_lock l(mu_);
+  timeout_in_ms_ = ms;
+}
+
 }  // end namespace tensorflow