Searched refs:collective_type (Results 1 – 2 of 2) sorted by relevance
397 CollectiveType collective_type, in AddParticipant() argument407 data_type, collective_type, reduction_op, context.num_local_devices, in AddParticipant()433 if (collective->status.ok() && collective->type != collective_type) { in AddParticipant()436 collective->type, " but now got type ", collective_type); in AddParticipant()
197 const Context& context, CollectiveType collective_type,