Method device

tensorflow/tensorflow/python/framework/ops.py:4308–4376 · view source on GitHub ↗

Returns a context manager that specifies the default device to use. The `device_name_or_function` argument may either be a device name string, a device function, or None: * If it is a device name string, all operations constructed in this context will be assigned to the device wi

(self, device_name_or_function)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

testNodeDefArgsMethod · 0.95

testDevicePartialStringMethod · 0.95

testDeviceFullMethod · 0.95

testNestingMethod · 0.95

testNestingStringMethod · 0.95

testNestingOverrideGpuCpuMethod · 0.95

testNestingWithMergeDeviceFunctionMethod · 0.95

testNestingWithDeviceStringsMethod · 0.95

testNestingWithDeviceStringWildcardMethod · 0.95

testNestingErrorGraphMethod · 0.95

testNoneClearsDefaultMethod · 0.95

Calls 3

_add_device_to_stackMethod · 0.95

peek_top_objMethod · 0.80

pop_objMethod · 0.80

Tested by 15

testNodeDefArgsMethod · 0.76

testDevicePartialStringMethod · 0.76

testDeviceFullMethod · 0.76

testNestingMethod · 0.76

testNestingStringMethod · 0.76

testNestingOverrideGpuCpuMethod · 0.76

testNestingWithMergeDeviceFunctionMethod · 0.76

testNestingWithDeviceStringsMethod · 0.76

testNestingWithDeviceStringWildcardMethod · 0.76

testNestingErrorGraphMethod · 0.76

testNoneClearsDefaultMethod · 0.76