Searched refs:peer_gpu_ids (Results 1 – 5 of 5) sorted by relevance
86 size_t total_bytes, const std::vector<TfGpuId>& peer_gpu_ids) { in CreateSubAllocator() argument106 platform_peer_gpu_ids.reserve(peer_gpu_ids.size()); in CreateSubAllocator()107 for (const TfGpuId tf_gpu_id : peer_gpu_ids) { in CreateSubAllocator()138 const std::vector<TfGpuId>& peer_gpu_ids) { in GetGPUAllocator() argument168 total_bytes, peer_gpu_ids); in GetGPUAllocator()
66 const std::vector<PlatformGpuId>& peer_gpu_ids) { in Create() argument68 access_gpu_handles.reserve(peer_gpu_ids.size() + 1); in Create()75 for (const auto& peer_id : peer_gpu_ids) { in Create()
49 const std::vector<PlatformGpuId>& peer_gpu_ids);
86 const std::vector<TfGpuId>& peer_gpu_ids);
1422 std::vector<TfGpuId> peer_gpu_ids; in CreateGPUDevice() local1423 peer_gpu_ids.reserve(num_tf_gpus); in CreateGPUDevice()1427 peer_gpu_ids.push_back(peer_tf_gpu_id); in CreateGPUDevice()1433 options.config.gpu_options(), tf_gpu_id, memory_limit, peer_gpu_ids); in CreateGPUDevice()