Home
last modified time | relevance | path

Searched refs:merge_device (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Ddevice_test.py44 with ops.device(device.merge_device("/device:GPU:0")):
47 with ops.device(device.merge_device("/job:worker")):
50 with ops.device(device.merge_device("/device:CPU:0")):
53 with ops.device(device.merge_device("/job:ps")):
Ddevice.py67 def merge_device(spec): function
Dimporter_test.py985 with ops.device(device.merge_device("/task:0")):
993 with ops.device(device.merge_device("/job:ps")):
1001 with ops.device(device.merge_device("/device:GPU:0")):
Dops_test.py1525 with g.device(pydev.merge_device("/device:GPU:0")):
1527 with g.device(pydev.merge_device("/job:worker")):
1529 with g.device(pydev.merge_device("/device:CPU:0")):
1531 with g.device(pydev.merge_device("/job:ps")):
1533 with g.device(pydev.merge_device(None)):
1666 with g.device(pydev.merge_device("/job:ps")): # Will be overwritten.
1672 with g.device(pydev.merge_device("/job:ps")):
Dops.py141 self.function = pydev.merge_device(device_name_or_function)
/external/tensorflow/tensorflow/python/eager/
Dfunction.py2608 device_functions = (pydev.merge_device(ctx.device_name),)