Searched refs:replica_groups_size (Results 1 – 1 of 1) sorted by relevance
1136 int32 replica_groups_size = replica_groups.size(); in HandleAllReduceMultipleReplica() local1180 /*replica_groups_size=*/b_.getInt32(replica_groups_size), in HandleAllReduceMultipleReplica()1217 int32 replica_groups_size = replica_groups.size(); in HandleAllToAll() local1252 /*replica_groups_size=*/b_.getInt32(replica_groups_size), in HandleAllToAll()