Searched defs:tf_gpu_id (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_manager.cc | 37 Status Insert(TfGpuId tf_gpu_id, PlatformGpuId platform_gpu_id) in Insert()
60 bool Find(TfGpuId tf_gpu_id, PlatformGpuId* platform_gpu_id) const in Find()
86 Status GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId tf_gpu_id, in InsertTfPlatformGpuIdPair()
91 Status GpuIdManager::TfToPlatformGpuId(TfGpuId tf_gpu_id, in TfToPlatformGpuId()
|
| D | gpu_id_utils.h | 42 TfGpuId tf_gpu_id) { in ExecutorForTfGpuId()
50 static void CheckValidTfGpuId(TfGpuId tf_gpu_id) { in CheckValidTfGpuId()
|
| D | gpu_process_state.cc | 70 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU()
81 TfGpuId tf_gpu_id, in GetGPUAllocator()
164 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()
|
| D | gpu_device_factory.cc | 32 TfGpuId tf_gpu_id, const string& physical_device_desc, in GPUDevice()
65 const DeviceLocality& locality, TfGpuId tf_gpu_id, in CreateGPUDevice()
|
| D | gpu_device.cc | 121 TfGpuId tf_gpu_id, ::tensorflow::Allocator* alloc, in Reinitialize()
237 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate()
300 TfGpuId tf_gpu_id, in BaseGPUDevice()
754 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize()
1133 TfGpuId tf_gpu_id(next_tf_gpu_id); in CreateDevices() local
1148 TfGpuId tf_gpu_id(di); in CreateDevices() local
1183 const SessionOptions& options, const string& name_prefix, TfGpuId tf_gpu_id, in CreateGPUDevice()
1283 for (TfGpuId tf_gpu_id : all_tf_gpu_ids) { in GetDeviceLocalities() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.cc | 159 TfGpuId tf_gpu_id(device.id); in GetDeviceInfo() local
|
| D | single_machine.cc | 95 TfGpuId tf_gpu_id(parsed.id); in Provision() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 243 TfGpuId tf_gpu_id(parsed.id); in GetDeviceInfo() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 738 TfGpuId tf_gpu_id(tf_gpu_id_value); in GetDeviceAndAllocator() local
|