Searched defs:local_rank (Results 1 – 2 of 2) sorted by relevance
125 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeReductionTestCase() local219 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in MakeBroadcastTestCase() local260 for (int local_rank = 0; local_rank < test_case->num_ranks_per_node; in VerifyResults() local331 for (int local_rank = 0; local_rank < num_ranks_per_node; in RunMultiNodeAllReduceTest() local376 for (int local_rank = 0; local_rank < num_ranks_per_node; ++local_rank) { in RunMultiNodeBroadcastTest() local421 static int GlobalRank(int num_ranks_per_node, int node, int local_rank) { in GlobalRank()426 int local_rank) { in GetDevice()
281 int local_rank; member