Home
last modified time | relevance | path

Searched refs:TfToPlatformGpuId (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_id_manager.h33 static Status TfToPlatformGpuId(TfDeviceId tf_gpu_id,
Dgpu_id_manager.cc29 Status GpuIdManager::TfToPlatformGpuId(TfDeviceId tf_gpu_id, in TfToPlatformGpuId() function in tensorflow::GpuIdManager
Dgpu_process_state.cc109 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CreateSubAllocator()
160 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetGPUAllocator()
Dgpu_device.cc136 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in Reinitialize()
478 GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in Init()
1412 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CreateGPUDevice()
1520 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetDeviceLocalities()
1555 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()
1572 GpuIdManager::TfToPlatformGpuId(tf_gpu_dst, &platform_gpu_dst)); in GetDeviceLocalities()
Dgpu_device.h91 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in gpu_id()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dutils.cc161 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
Dsingle_machine.cc97 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in Provision()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc246 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc109 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetFirstValidDeviceId()