diff options
author | Akshay Modi <nareshmodi@google.com> | 2018-08-16 14:44:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-16 14:50:47 -0700 |
commit | 2a08dd7f020138f5b79af188504937321f4d542d (patch) | |
tree | 98c04581e369e312fe0e234224c6c641dcd9215e /tensorflow/contrib/eager | |
parent | d43820b9eff0cc863de2bbfb142afe92bf5afd00 (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.py | 4 |
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 |