diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-07-26 10:20:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-26 10:27:55 -0700 |
| commit | d6d169d511f83fc677288e67c65cf354d31341bb (patch) | |
| tree | ccba2193c706e19be68d0eade2431c5cbcb55c03 | |
| parent | 883e8b6863511b46d7985e9ff8d1809ffe2a1bc0 (diff) | |
Allow setting server_def on tf.enable_eager_execution
PiperOrigin-RevId: 206179345
| -rw-r--r-- | tensorflow/contrib/eager/python/tfe.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/eager/python/tfe.py b/tensorflow/contrib/eager/python/tfe.py index ca6430253b..2f0ab616e4 100644 --- a/tensorflow/contrib/eager/python/tfe.py +++ b/tensorflow/contrib/eager/python/tfe.py @@ -34,6 +34,7 @@ To use, at program startup, call `tfe.enable_eager_execution()`. @@run @@enable_eager_execution +@@enable_remote_eager_execution @@custom_gradient @@ -114,6 +115,7 @@ from tensorflow.python.eager.execution_callbacks import inf_nan_callback from tensorflow.python.eager.execution_callbacks import nan_callback from tensorflow.python.eager.execution_callbacks import seterr from tensorflow.python.framework.ops import enable_eager_execution +from tensorflow.python.framework.ops import enable_eager_execution_internal as enable_remote_eager_execution from tensorflow.python.framework.ops import eager_run as run from tensorflow.python.framework.test_util import run_in_graph_and_eager_modes as run_test_in_graph_and_eager_modes from tensorflow.python.framework.test_util import run_all_in_graph_and_eager_modes as run_all_tests_in_graph_and_eager_modes |