Searched defs:peer_gpu_ids (Results 1 – 3 of 3) sorted by relevance
101 size_t total_bytes, const std::vector<TfDeviceId>& peer_gpu_ids) { in CreateSubAllocator()153 const std::vector<TfDeviceId>& peer_gpu_ids) { in GetGPUAllocator()
66 const std::vector<PlatformDeviceId>& peer_gpu_ids) { in Create()
1582 std::vector<TfDeviceId> peer_gpu_ids; in CreateGPUDevice() local