aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/eager
diff options
context:
space:
mode:
authorGravatar Akshay Modi <nareshmodi@google.com>2018-08-16 14:44:37 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-16 14:50:47 -0700
commit2a08dd7f020138f5b79af188504937321f4d542d (patch)
tree98c04581e369e312fe0e234224c6c641dcd9215e /tensorflow/contrib/eager
parentd43820b9eff0cc863de2bbfb142afe92bf5afd00 (diff)
Allow tfe.set_server_def to enable remote eager execution.
Allows a change any time within process. PiperOrigin-RevId: 209051119
Diffstat (limited to 'tensorflow/contrib/eager')
-rw-r--r--tensorflow/contrib/eager/python/tfe.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/eager/python/tfe.py b/tensorflow/contrib/eager/python/tfe.py
index de11d00a1a..4dfd083443 100644
--- a/tensorflow/contrib/eager/python/tfe.py
+++ b/tensorflow/contrib/eager/python/tfe.py
@@ -16,7 +16,7 @@
EXPERIMENTAL: APIs here are unstable and likely to change without notice.
-To use, at program startup, call `tfe.enable_eager_execution()`.
+To use, at program startup, call `tf.enable_eager_execution()`.
@@metrics
@@ -67,6 +67,7 @@ To use, at program startup, call `tfe.enable_eager_execution()`.
@@execution_mode
@@async_wait
@@async_clear_error
+@@set_server_def
@@run_test_in_graph_and_eager_modes
@@run_all_tests_in_graph_and_eager_modes
@@ -110,6 +111,7 @@ from tensorflow.python.eager.context import async_clear_error
from tensorflow.python.eager.context import SYNC
from tensorflow.python.eager.context import ASYNC
from tensorflow.python.eager.context import num_gpus
+from tensorflow.python.eager.context import set_server_def
from tensorflow.python.eager.execution_callbacks import add_execution_callback
from tensorflow.python.eager.execution_callbacks import clear_execution_callbacks
from tensorflow.python.eager.execution_callbacks import inf_callback