| /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() argument
42 result = id_map_.insert({tf_gpu_id.value(), platform_gpu_id.value()}); in Insert()
46 "TensorFlow device (GPU:", tf_gpu_id.value(), in Insert()
60 bool Find(TfGpuId tf_gpu_id, PlatformGpuId* platform_gpu_id) const in Find() argument
63 auto result = id_map_.find(tf_gpu_id.value()); in Find()
86 Status GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId tf_gpu_id, in InsertTfPlatformGpuIdPair() argument
88 return TfToPlatformGpuIdMap::singleton()->Insert(tf_gpu_id, platform_gpu_id); in InsertTfPlatformGpuIdPair()
91 Status GpuIdManager::TfToPlatformGpuId(TfGpuId tf_gpu_id, in TfToPlatformGpuId() argument
93 if (TfToPlatformGpuIdMap::singleton()->Find(tf_gpu_id, platform_gpu_id)) { in TfToPlatformGpuId()
96 return errors::NotFound("TensorFlow device GPU:", tf_gpu_id.value(), in TfToPlatformGpuId()
|
| D | gpu_process_state.cc | 70 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU() argument
73 GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id).ValueOrDie(); in BusIdForGPU()
81 TfGpuId tf_gpu_id, in GetGPUAllocator() argument
87 GpuIdUtil::CheckValidTfGpuId(tf_gpu_id); in GetGPUAllocator()
89 if (tf_gpu_id.value() >= static_cast<int64>(gpu_allocators_.size())) { in GetGPUAllocator()
90 gpu_allocators_.resize(tf_gpu_id.value() + 1); in GetGPUAllocator()
93 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator()
102 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in GetGPUAllocator()
103 int bus_id = BusIdForGPU(tf_gpu_id); in GetGPUAllocator()
116 strings::StrCat("GPU_", tf_gpu_id.value(), "_bfc")); in GetGPUAllocator()
[all …]
|
| D | gpu_id_utils.h | 42 TfGpuId tf_gpu_id) { in ExecutorForTfGpuId() argument
45 GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in ExecutorForTfGpuId()
50 static void CheckValidTfGpuId(TfGpuId tf_gpu_id) { in CheckValidTfGpuId() argument
52 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in CheckValidTfGpuId()
56 << " TF GPU id: " << tf_gpu_id in CheckValidTfGpuId()
|
| D | gpu_device.cc | 121 TfGpuId tf_gpu_id, ::tensorflow::Allocator* alloc, in Reinitialize() argument
134 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id)); in Reinitialize()
237 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate() argument
243 &streams_[key_type(tf_gpu_id.value(), stream_group_within_gpu)]; in GetOrCreate()
300 TfGpuId tf_gpu_id, in BaseGPUDevice() argument
310 tf_gpu_id_(tf_gpu_id), in BaseGPUDevice()
325 GPUProcessState::singleton()->GPUAllocatorCounter(tf_gpu_id); in BaseGPUDevice()
754 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize() argument
756 stream_device_.Reinitialize(context, gpu_stream, tf_gpu_id, base_allocator, in Reinitialize()
1133 TfGpuId tf_gpu_id(next_tf_gpu_id); in CreateDevices() local
[all …]
|
| D | gpu_device_factory.cc | 32 TfGpuId tf_gpu_id, const string& physical_device_desc, in GPUDevice() argument
34 : BaseGPUDevice(options, name, memory_limit, locality, tf_gpu_id, in GPUDevice()
65 const DeviceLocality& locality, TfGpuId tf_gpu_id, in CreateGPUDevice() argument
69 tf_gpu_id, physical_device_desc, in CreateGPUDevice()
|
| D | gpu_process_state.h | 84 TfGpuId tf_gpu_id, size_t total_bytes);
111 virtual int BusIdForGPU(TfGpuId tf_gpu_id);
113 SharedCounter* GPUAllocatorCounter(TfGpuId tf_gpu_id);
|
| D | gpu_id_manager.h | 29 static Status InsertTfPlatformGpuIdPair(TfGpuId tf_gpu_id,
33 static Status TfToPlatformGpuId(TfGpuId tf_gpu_id,
|
| D | gpu_device.h | 56 TfGpuId tf_gpu_id, const string& physical_device_desc,
301 const string& name_prefix, TfGpuId tf_gpu_id,
307 const DeviceLocality& dev_locality, TfGpuId tf_gpu_id,
|
| /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/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 738 TfGpuId tf_gpu_id(tf_gpu_id_value); in GetDeviceAndAllocator() local
740 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceAndAllocator()
742 VLOG(1) << "Found TF GPU " << tf_gpu_id.value() << " at cuda device " in GetDeviceAndAllocator()
750 gpu_options, tf_gpu_id, 1); in GetDeviceAndAllocator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 243 TfGpuId tf_gpu_id(parsed.id); in GetDeviceInfo() local
245 Status s = GpuIdManager::TfToPlatformGpuId(tf_gpu_id, &platform_gpu_id); in GetDeviceInfo()
|