message: eager: Expose tfe.run_test_in_graph_and_eager_modes decorator. Useful when writing tests of libraries.
diff: @@ -50,6 +50,8 @@ To use, at program startup, call `tfe.enable_eager_execution()`.
  @@in_eager_mode
  @@in_graph_mode
  +
  +@@run_test_in_graph_and_eager_modes
  """
  from __future__ import absolute_import
  @@ -60,12 +62,10 @@ from __future__ import print_function
  # pylint:disable=g-bad-import-order,g-import-not-at-top,unused-import
  #
  from tensorflow.contrib.eager.python.datasets import Iterator
  -from tensorflow.contrib.eager.python.saver import Saver
  from tensorflow.contrib.eager.python.saver import restore_variables_on_create
  +from tensorflow.contrib.eager.python.saver import Saver
  from tensorflow.contrib.eager.python.summary_writer import SummaryWriter
  -from tensorflow.python.util.all_util import remove_undocumented
  from tensorflow.python.eager import backprop
  -from tensorflow.python.eager.custom_gradient import custom_gradient
  from tensorflow.python.eager import function
  from tensorflow.python.eager.context import enable_eager_execution
  from tensorflow.python.eager.context import in_eager_mode
  @@ -74,13 +74,16 @@ from tensorflow.python.eager.context import list_devices
  from tensorflow.python.eager.context import num_gpus
  from tensorflow.python.eager.context import run
  from tensorflow.python.eager.core import enable_tracing
  +from tensorflow.python.eager.custom_gradient import custom_gradient
  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
  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.test_util import run_in_graph_and_eager_modes as run_test_in_graph_and_eager_modes
  from tensorflow.python.ops.resource_variable_ops import ResourceVariable as Variable
  +from tensorflow.python.util.all_util import remove_undocumented
  defun = function.defun
  implicit_gradients = backprop.implicit_grad
  