Searched refs:global_rank (Results 1 – 4 of 4) sorted by relevance
60 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()67 global_rank(global_rank), in Participant()103 const int global_rank; member
438 const int global_rank = node * num_ranks_per_node + local_rank; in TYPED_TEST() local441 &test_case->ins[global_rank], &test_case->outs[global_rank], in TYPED_TEST()442 global_rank, this->CreateDoneCallback(test_case.get())); in TYPED_TEST()449 << global_rank; in TYPED_TEST()
316 const int rank = collective->participants[i]->global_rank >= 0 in GetCommunicator()317 ? collective->participants[i]->global_rank in GetCommunicator()507 const int rank = p->global_rank >= 0 ? p->global_rank : i; in RunCollective()
152 int global_rank; member171 dr->global_rank = 0; // Will be populated later by EstablishGlobalRank. in BuildDevRecs()311 it.second.global_rank = it.second.local_rank + next_rank; in EstablishGlobalRank()474 new_device_names[dr.global_rank] = in CompleteDefaultRanking()476 new_task_names[dr.global_rank] = in CompleteDefaultRanking()