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

Method device

tensorflow/python/eager/context.py:234–280 · view source on GitHub ↗

Context-manager to force placement of operations and Tensors on a device. Args: name: Name of the device or None to get default placement. Yields: Nothing. Raises: ValueError: If name is not a string or is an invalid device name.

(self, name)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

testContextMethod · 0.95

deviceFunction · 0.45

testInt32CPUDefaultMethod · 0.45

testBadConstructorArgsMethod · 0.45

testStringTensorOnGPUMethod · 0.45

testTensorPlacementMethod · 0.45

testOperationWithNoInputsRunsOnDeviceMethod · 0.45

testInvalidDeviceMethod · 0.45

testBenchmarksMethod · 0.45

fnMethod · 0.45

testGPUImplicitGradMethod · 0.45

fMethod · 0.45

Calls 4

from_stringMethod · 0.80

merge_fromMethod · 0.80

to_stringMethod · 0.80

copyMethod · 0.45

Tested by 13

testContextMethod · 0.76

testInt32CPUDefaultMethod · 0.36

testBadConstructorArgsMethod · 0.36

testStringTensorOnGPUMethod · 0.36

testTensorPlacementMethod · 0.36

testOperationWithNoInputsRunsOnDeviceMethod · 0.36

testInvalidDeviceMethod · 0.36

testBenchmarksMethod · 0.36

fnMethod · 0.36

testGPUImplicitGradMethod · 0.36

fMethod · 0.36

testTensorCopyGPU2CPU2GPUMethod · 0.36