| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 71 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU() argument
74 DEVICE_GPU, GPUMachineManager(), tf_gpu_id) in BusIdForGPU()
107 for (const TfGpuId tf_gpu_id : peer_gpu_ids) { in CreateSubAllocator()
109 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CreateSubAllocator()
137 const GPUOptions& options, TfGpuId tf_gpu_id, size_t total_bytes, in GetGPUAllocator() argument
145 tf_gpu_id); in GetGPUAllocator()
147 if (tf_gpu_id.value() >= static_cast<int64>(gpu_allocators_.size())) { in GetGPUAllocator()
148 gpu_allocators_.resize(tf_gpu_id.value() + 1); in GetGPUAllocator()
151 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator()
160 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetGPUAllocator()
[all …]
|
| D | gpu_id_manager.cc | 24 TfDeviceId tf_gpu_id, PlatformDeviceId platform_gpu_id) { in InsertTfPlatformGpuIdPair() argument
25 return DeviceIdManager::InsertTfPlatformDeviceIdPair(DEVICE_GPU, tf_gpu_id, in InsertTfPlatformGpuIdPair()
29 Status GpuIdManager::TfToPlatformGpuId(TfDeviceId tf_gpu_id, in TfToPlatformGpuId() argument
31 return DeviceIdManager::TfToPlatformDeviceId(DEVICE_GPU, tf_gpu_id, in TfToPlatformGpuId()
|
| D | gpu_device.cc | 123 TfGpuId tf_gpu_id, ::tensorflow::Allocator* alloc, in Reinitialize() argument
136 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in Reinitialize()
239 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate() argument
245 &streams_[key_type(tf_gpu_id.value(), stream_group_within_gpu)]; in GetOrCreate()
247 int priority = GetPriority(tf_gpu_id.value(), options); in GetOrCreate()
342 int GetPriority(int tf_gpu_id, const GPUOptions& options) { in GetPriority() argument
343 int id = tf_gpu_id; in GetPriority()
381 TfGpuId tf_gpu_id, in BaseGPUDevice() argument
391 tf_gpu_id_(tf_gpu_id), in BaseGPUDevice()
766 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize() argument
[all …]
|
| D | gpu_device_factory.cc | 33 TfGpuId tf_gpu_id, const string& physical_device_desc, in GPUDevice() argument
35 : BaseGPUDevice(options, name, memory_limit, locality, tf_gpu_id, in GPUDevice()
66 const DeviceLocality& locality, TfGpuId tf_gpu_id, in CreateGPUDevice() argument
70 tf_gpu_id, physical_device_desc, in CreateGPUDevice()
|
| D | gpu_id_manager.h | 29 static Status InsertTfPlatformGpuIdPair(TfDeviceId tf_gpu_id,
33 static Status TfToPlatformGpuId(TfDeviceId tf_gpu_id,
|
| D | gpu_process_state.h | 85 TfGpuId tf_gpu_id, size_t total_bytes,
118 virtual int BusIdForGPU(TfGpuId tf_gpu_id);
120 SharedCounter* GPUAllocatorCounter(TfGpuId tf_gpu_id);
|
| D | gpu_device.h | 56 TfGpuId tf_gpu_id, const std::string& physical_device_desc,
355 const std::string& name_prefix, TfGpuId tf_gpu_id,
362 const DeviceLocality& dev_locality, TfGpuId tf_gpu_id,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 107 TfGpuId tf_gpu_id(tf_gpu_id_value); in GetFirstValidDeviceId() local
109 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetFirstValidDeviceId()
111 VLOG(1) << "Found TF GPU " << tf_gpu_id.value() << " at cuda device " in GetFirstValidDeviceId()
113 return std::make_pair(tf_gpu_id, platform_gpu_id); in GetFirstValidDeviceId()
269 TfGpuId tf_gpu_id; in GetEngineInfo() local
271 std::tie(tf_gpu_id, platform_gpu_id) = GetFirstValidDeviceId(); in GetEngineInfo()
272 if (tf_gpu_id.value() >= 0) { in GetEngineInfo()
276 parsed_name.id = tf_gpu_id.value(); in GetEngineInfo()
638 TfGpuId tf_gpu_id; in GetDeviceAndAllocator() local
640 std::tie(tf_gpu_id, platform_gpu_id) = GetFirstValidDeviceId(); in GetDeviceAndAllocator()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.cc | 159 TfGpuId tf_gpu_id(device.id); in GetDeviceInfo() local
161 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
|
| D | single_machine.cc | 95 TfGpuId tf_gpu_id(parsed.id); in Provision() local
97 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in Provision()
100 tf_gpu_id.value(), ": ", s.ToString()); in Provision()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 244 TfGpuId tf_gpu_id(parsed.id); in GetDeviceInfo() local
246 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
|