Searched defs:tf_gpu_id (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_manager.cc | 24 TfDeviceId tf_gpu_id, PlatformDeviceId platform_gpu_id) { in InsertTfPlatformGpuIdPair()
29 Status GpuIdManager::TfToPlatformGpuId(TfDeviceId tf_gpu_id, in TfToPlatformGpuId()
|
| D | gpu_process_state.cc | 71 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU()
137 const GPUOptions& options, TfGpuId tf_gpu_id, size_t total_bytes, in GetGPUAllocator()
221 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()
|
| D | gpu_device_factory.cc | 33 TfGpuId tf_gpu_id, const string& physical_device_desc, in GPUDevice()
66 const DeviceLocality& locality, TfGpuId tf_gpu_id, in CreateGPUDevice()
|
| D | gpu_device.cc | 123 TfGpuId tf_gpu_id, ::tensorflow::Allocator* alloc, in Reinitialize()
239 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate()
342 int GetPriority(int tf_gpu_id, const GPUOptions& options) { in GetPriority()
381 TfGpuId tf_gpu_id, in BaseGPUDevice()
766 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize()
1351 TfGpuId tf_gpu_id(next_tf_gpu_id); in CreateDevices() local
1366 TfGpuId tf_gpu_id(di); in CreateDevices() local
1402 const SessionOptions& options, const string& name_prefix, TfGpuId tf_gpu_id, in CreateGPUDevice()
1517 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/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 107 TfGpuId tf_gpu_id(tf_gpu_id_value); in GetFirstValidDeviceId() local
269 TfGpuId tf_gpu_id; in GetEngineInfo() local
638 TfGpuId tf_gpu_id; in GetDeviceAndAllocator() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 244 TfGpuId tf_gpu_id(parsed.id); in GetDeviceInfo() local
|