Function device

tensorflow/tensorflow/python/framework/ops.py:5109–5141 · view source on GitHub ↗

Wrapper for `Graph.device()` using the default graph. See `tf.Graph.device` for more details. Args: device_name_or_function: The device name or function to use in the context. Returns: A context manager that specifies the default device to use for newly created ops. Raises:

(device_name_or_function)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

_NodeDefFunction · 0.70

device_v2Function · 0.70

_colocate_with_for_gradientFunction · 0.70

Calls 5

executing_eagerly_outside_functionsFunction · 0.85

combinedFunction · 0.85

get_default_graphFunction · 0.85

executing_eagerlyMethod · 0.80

deviceMethod · 0.45

Tested by

no test coverage detected