Searched refs:platform_device_id (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_utils.h | 35 se::Platform* device_manager, PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId() argument
36 return device_manager->ExecutorForDevice(platform_device_id.value()); in ExecutorForPlatformDeviceId()
41 PlatformDeviceId platform_device_id; in ExecutorForTfDeviceId() local
43 type, tf_device_id, &platform_device_id)); in ExecutorForTfDeviceId()
44 return ExecutorForPlatformDeviceId(device_manager, platform_device_id); in ExecutorForTfDeviceId()
52 PlatformDeviceId platform_device_id; in CheckValidTfDeviceId() local
54 &platform_device_id)); in CheckValidTfDeviceId()
56 CHECK_LT(platform_device_id.value(), visible_device_count) in CheckValidTfDeviceId()
59 << " id: " << platform_device_id in CheckValidTfDeviceId()
|
| D | device_id_manager.cc | 38 PlatformDeviceId platform_device_id) TF_LOCKS_EXCLUDED(mu_) { in Insert() argument
45 {tf_device_id.value(), platform_device_id.value()}); in Insert()
47 if (!result.second && platform_device_id.value() != result.first->second) { in Insert()
50 ") is being mapped to multiple devices (", platform_device_id.value(), in Insert()
63 PlatformDeviceId* platform_device_id) const TF_LOCKS_EXCLUDED(mu_) { in Find()
71 *platform_device_id = id_map_iter->second; in Find()
99 PlatformDeviceId platform_device_id) { in InsertTfPlatformDeviceIdPair() argument
101 platform_device_id); in InsertTfPlatformDeviceIdPair()
106 PlatformDeviceId* platform_device_id) { in TfToPlatformDeviceId() argument
108 platform_device_id)) { in TfToPlatformDeviceId()
|
| D | device_id_manager_test.cc | 27 PlatformDeviceId platform_device_id; in TfToPlatformDeviceId() local
29 DeviceIdManager::TfToPlatformDeviceId(type, tf, &platform_device_id)); in TfToPlatformDeviceId()
30 return platform_device_id; in TfToPlatformDeviceId()
|
| D | device_id_manager.h | 32 PlatformDeviceId platform_device_id);
38 PlatformDeviceId* platform_device_id);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 41 PlatformDeviceId platform_device_id) { in ExecutorForPlatformGpuId() argument
43 platform_device_id) in ExecutorForPlatformGpuId()
|