Searched refs:set_visible_devices (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | config.py | 493 def set_visible_devices(devices, device_type=None): function 523 context.context().set_visible_devices(devices, device_type)
|
D | config_test.py | 445 config.set_visible_devices(cpus[0]) 458 config.set_visible_devices(gpus) 461 config.set_visible_devices(cpus[0]) 506 config.set_visible_devices(gpus[1], 'GPU') 661 config.set_visible_devices(gpus[0], 'GPU') 666 config.set_visible_devices(gpus[1], 'GPU') 671 config.set_visible_devices(gpus, 'GPU')
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.config.pbtxt | 80 name: "set_visible_devices"
|
D | tensorflow.config.experimental.pbtxt | 88 name: "set_visible_devices"
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.config.pbtxt | 80 name: "set_visible_devices"
|
D | tensorflow.config.experimental.pbtxt | 88 name: "set_visible_devices"
|
/external/tensorflow/tensorflow/tools/docs/ |
D | tf_doctest.py | 151 tf.config.set_visible_devices(available_gpus[:required_gpus])
|
/external/tensorflow/tensorflow/python/eager/ |
D | context.py | 1361 self.set_visible_devices([], "CPU") 1390 self.set_visible_devices(visible_gpus, "GPU") 1411 def set_visible_devices(self, devices, device_type=None): member in Context
|
/external/tensorflow/ |
D | RELEASE.md | 1566 …ist_physical_devices`, `tf.config.get_visible_devices`, `tf.config.set_visible_devices`, `tf.confi…
|