Searched refs:register_custom_device (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | custom_device_test.py | 30 context.register_custom_device(device, device_name, device_info)
|
| D | context.py | 1313 def register_custom_device(self, device_capsule, device_name, member in Context
2703 def register_custom_device(device_capsule, device_name, device_info_capsule): function
2725 context().register_custom_device(device_capsule, device_name,
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device.py | 78 context.register_custom_device(device, self._name, device_info)
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | dtensor_device.py | 71 context.register_custom_device(device, self.name, device_info)
|