Method _map_py_func

tensorflow/python/data/kernel_tests/map_test.py:637–640

(x)

    set_event_during_invocation = 5

    def _map_py_func(x):
      if x == set_event_during_invocation:
        ev.set()
      return x * x

    def _map_fn(x):
      return script_ops.py_func(_map_py_func, [x], x.dtype)