Function decorated

tensorflow/python/framework/test_util.py:980–997

(self, run_mode, *args, **kwargs)

                                    ("_function", "function"))
    @functools.wraps(f)
    def decorated(self, run_mode, *args, **kwargs):
      if run_mode == "v1_graph":
        with ops.Graph().as_default():
          f(self, *args, **kwargs)
      elif run_mode == "function":

        @def_function.function
        def function_in_eager():
          f(self, *args, **kwargs)

        # Create a new graph for the eagerly executed version of this test for
        # better isolation.
        graph_for_eager_test = ops.Graph()
        with graph_for_eager_test.as_default(), context.eager_mode():
          function_in_eager()
        ops.dismantle_graph(graph_for_eager_test)
      else:
        return ValueError("Unknown run mode %s" % run_mode)

    return decorated