Searched defs:Collective (Results 1 – 2 of 2) sorted by relevance
162 struct NcclManager::Collective : public core::RefCounted { struct in tensorflow::NcclManager163 Collective(const string& collective_key_in, DataType data_type_in, in Collective() argument188 const string collective_key; // A unique key for debugging.189 const DataType data_type;190 const CollectiveType type;191 const ncclRedOp_t reduction_op; // applies when <type> is a reduction.192 const int num_local_devices; // devices local to this node193 const int num_global_devices; // devices across all nodes194 const bool single_node; // true if all devices are at one node195 const string communicator_key;[all …]
239 Collective, // An ArrayRef<Type> for all result types. enumerator