Searched refs:local_gathered (Results 1 – 1 of 1) sorted by relevance
32 def checkAllgather(self, num_ranks, all_gathered, local_gathered): argument35 loc_gat_ind = np.sort(local_gathered.indices)42 for i in range(len(local_gathered.indices)):48 for j in range(len(local_gathered.indices)):49 if local_gathered.indices[j] == all_index and not local_checked[j]:54 correct_output = local_gathered.values[loc_index][0]107 all_gathered, local_gathered = sess.run([all_gather, correct_gather])110 self.checkAllgather(num_ranks, all_gathered, local_gathered)