Function use_gpu

tensorflow/python/framework/test_util.py:1463–1466

Uses gpu when requested and available.

()

@contextlib.contextmanager
def use_gpu():
  """Uses gpu when requested and available."""
  with device(use_gpu=True):
    yield


@contextlib.contextmanager