ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


View source on GitHub

Wrapper for Graph.device() using the default graph.

See tf.Graph.device for more details.

device_name_or_function The device name or function to use in the context.

A context manager that specifies the default device to use for newly created ops.

RuntimeError If eager execution is enabled and a function is passed in.