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…
|