(self, device_name, device_spec)
| 726 | return self._thread_local_data.device_spec |
| 727 | |
| 728 | def _set_device(self, device_name, device_spec): |
| 729 | self._thread_local_data.device_name = device_name |
| 730 | self._thread_local_data.device_spec = device_spec |
| 731 | |
| 732 | def device(self, name): |
| 733 | """Context-manager to force placement of operations and Tensors on a device. |