Searched defs:scatter_group_size (Results 1 – 1 of 1) sorted by relevance
90 size_t scatter_group_size = std::max(num_local_devices, size_t(2)); in TryDecomposeReplicaGroup() local137 int scatter_group_size = in TryDecomposeReplicaGroups() local209 int scatter_group_size = in TryDecomposeAllReduce() local