Searched refs:TfGpuId (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.h | 85 TfGpuId tf_gpu_id, size_t total_bytes,
86 const std::vector<TfGpuId>& peer_gpu_ids);
118 virtual int BusIdForGPU(TfGpuId tf_gpu_id);
120 SharedCounter* GPUAllocatorCounter(TfGpuId tf_gpu_id);
|
| D | gpu_process_state.cc | 71 int GPUProcessState::BusIdForGPU(TfGpuId tf_gpu_id) { in BusIdForGPU()
86 size_t total_bytes, const std::vector<TfGpuId>& peer_gpu_ids) { in CreateSubAllocator()
107 for (const TfGpuId tf_gpu_id : peer_gpu_ids) { in CreateSubAllocator()
137 const GPUOptions& options, TfGpuId tf_gpu_id, size_t total_bytes, in GetGPUAllocator()
138 const std::vector<TfGpuId>& peer_gpu_ids) { in GetGPUAllocator()
221 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()
284 TfGpuId(i)) in GetGpuHostAllocator()
|
| D | gpu_device.h | 56 TfGpuId tf_gpu_id, const std::string& physical_device_desc,
149 TfGpuId tf_gpu_id_;
339 std::size_t operator()(const TfGpuId& id) const noexcept { in operator()
343 typedef std::unordered_map<TfGpuId, DeviceLocality, TfGpuIdHash> LocalityMap;
355 const std::string& name_prefix, TfGpuId tf_gpu_id,
362 const DeviceLocality& dev_locality, TfGpuId tf_gpu_id,
|
| 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()
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()
1366 TfGpuId tf_gpu_id(di); in CreateDevices()
1402 const SessionOptions& options, const string& name_prefix, TfGpuId tf_gpu_id, in CreateGPUDevice()
1422 std::vector<TfGpuId> peer_gpu_ids; in CreateGPUDevice()
1425 TfGpuId peer_tf_gpu_id(id); in CreateGPUDevice()
1512 std::vector<TfGpuId> all_tf_gpu_ids; in GetDeviceLocalities()
[all …]
|
| D | gpu_id.h | 24 typedef TfDeviceId TfGpuId; typedef
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils_test.cc | 101 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(0), PlatformGpuId(100))); in TEST()
107 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(1), PlatformGpuId(0))); in TEST()
|
| D | utils.cc | 159 TfGpuId tf_gpu_id(device.id); in GetDeviceInfo()
|
| D | single_machine.cc | 95 TfGpuId tf_gpu_id(parsed.id); in Provision()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 105 std::pair<TfGpuId, PlatformGpuId> GetFirstValidDeviceId() { in GetFirstValidDeviceId()
107 TfGpuId tf_gpu_id(tf_gpu_id_value); in GetFirstValidDeviceId()
117 return std::make_pair(TfGpuId(-1), PlatformGpuId(-1)); in GetFirstValidDeviceId()
269 TfGpuId tf_gpu_id; in GetEngineInfo()
638 TfGpuId tf_gpu_id; in GetDeviceAndAllocator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 244 TfGpuId tf_gpu_id(parsed.id); in GetDeviceInfo()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 152 GPUOptions(), TfGpuId(0), memory_limit, /*peer_gpu_ids=*/{}); in GPUDeviceTestHelper()
|