hub / github.com/benoitsteiner/tensorflow-opencl / device

Function device

tensorflow/python/framework/ops.py:4291–4310 · 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.

(device_name_or_function)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

_NodeDefFunction · 0.70

colocate_withFunction · 0.70

Calls 3

get_default_graphFunction · 0.85

in_graph_modeMethod · 0.80

deviceMethod · 0.45

Tested by

no test coverage detected