Helper to initialize devices.
(self)
| 448 | return self._rng.randint(0, _MAXINT32) |
| 449 | |
| 450 | def _initialize_logical_devices(self): |
| 451 | """Helper to initialize devices.""" |
| 452 | # Store list of devices |
| 453 | self._logical_devices = [] |
| 454 | self._context_devices = [] |
| 455 | device_list = pywrap_tensorflow.TFE_ContextListDevices( |
| 456 | self._context_handle) |
| 457 | try: |
| 458 | self._num_gpus = 0 |
| 459 | for i in range(pywrap_tensorflow.TF_DeviceListCount(device_list)): |
| 460 | dev_name = pywrap_tensorflow.TF_DeviceListName(device_list, i) |
| 461 | self._context_devices.append(pydev.canonical_name(dev_name)) |
| 462 | spec = pydev.DeviceSpec.from_string(dev_name) |
| 463 | self._logical_devices.append( |
| 464 | LogicalDevice(name=dev_name, device_type=spec.device_type)) |
| 465 | dev_type = pywrap_tensorflow.TF_DeviceListType(device_list, i) |
| 466 | if dev_type == "GPU": |
| 467 | self._num_gpus += 1 |
| 468 | |
| 469 | finally: |
| 470 | pywrap_tensorflow.TF_DeleteDeviceList(device_list) |
| 471 | |
| 472 | def ensure_initialized(self): |
| 473 | """Initialize handle and devices if not already done so.""" |
no test coverage detected