Searched refs:gpu_device_id (Results 1 – 2 of 2) sorted by relevance
59 EventMgr* event_mgr, int gpu_device_id, const Tensor* input, in Participant()64 gpu_device_id(gpu_device_id), in Participant()91 const int gpu_device_id; member
298 devices[i] = collective->participants[i]->gpu_device_id; in GetCommunicator()