Searched refs:TfToPlatformGpuId (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_manager_test.cc | 25 PlatformGpuId TfToPlatformGpuId(TfGpuId tf) { in TfToPlatformGpuId() function
27 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf, &platform_gpu_id)); in TfToPlatformGpuId()
35 EXPECT_EQ(value_0, TfToPlatformGpuId(key_0)); in TEST()
39 EXPECT_EQ(value_0, TfToPlatformGpuId(key_0)); in TEST()
45 EXPECT_EQ(value_1, TfToPlatformGpuId(key_1)); in TEST()
50 EXPECT_EQ(value_1, TfToPlatformGpuId(key_2)); in TEST()
56 ASSERT_FALSE(GpuIdManager::TfToPlatformGpuId(TfGpuId(100), &value_0).ok()); in TEST()
|
| D | gpu_id_utils.h | 45 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in ExecutorForTfGpuId()
52 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CheckValidTfGpuId()
|
| D | gpu_id_manager.h | 33 static Status TfToPlatformGpuId(TfGpuId tf_gpu_id,
|
| D | gpu_id_manager.cc | 91 Status GpuIdManager::TfToPlatformGpuId(TfGpuId tf_gpu_id, in TfToPlatformGpuId() function in tensorflow::GpuIdManager
|
| D | gpu_device.cc | 134 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in Reinitialize()
400 GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in Init()
1192 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CreateGPUDevice()
1286 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetDeviceLocalities()
1317 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()
1334 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()
|
| D | gpu_device.h | 104 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in gpu_id()
|
| D | gpu_process_state.cc | 102 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetGPUAllocator()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.cc | 161 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
|
| D | single_machine.cc | 97 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in Provision()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 245 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 740 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceAndAllocator()
|