Searched refs:InsertTfPlatformGpuIdPair (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_manager_test.cc | 34 TF_ASSERT_OK(GpuIdManager::InsertTfPlatformGpuIdPair(key_0, value_0)); in TEST()
38 TF_ASSERT_OK(GpuIdManager::InsertTfPlatformGpuIdPair(key_0, value_0)); in TEST()
44 TF_ASSERT_OK(GpuIdManager::InsertTfPlatformGpuIdPair(key_1, value_1)); in TEST()
49 TF_ASSERT_OK(GpuIdManager::InsertTfPlatformGpuIdPair(key_2, value_1)); in TEST()
53 ASSERT_FALSE(GpuIdManager::InsertTfPlatformGpuIdPair(key_2, value_0).ok()); in TEST()
|
| D | gpu_id_manager.h | 29 static Status InsertTfPlatformGpuIdPair(TfGpuId tf_gpu_id,
|
| D | gpu_id_manager.cc | 86 Status GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId tf_gpu_id, in InsertTfPlatformGpuIdPair() function in tensorflow::GpuIdManager
|
| D | gpu_device.cc | 1136 GpuIdManager::InsertTfPlatformGpuIdPair(tf_gpu_id, platform_gpu_id)); in CreateDevices()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils_test.cc | 87 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(0), PlatformGpuId(100))); in TEST()
93 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(1), PlatformGpuId(0))); in TEST()
|