Searched refs:num_global_ranks (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 311 const int num_global_ranks = num_nodes * num_ranks_per_node; in RunMultiNodeAllReduceTest() local
328 auto node_fn = [this, node, num_ranks_per_node, num_global_ranks, in RunMultiNodeAllReduceTest()
344 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeAllReduceTest()
366 const int num_global_ranks = num_nodes * num_ranks_per_node; in RunMultiNodeBroadcastTest() local
379 auto rank_fn = [this, node, num_ranks_per_node, num_global_ranks, in RunMultiNodeBroadcastTest()
399 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeBroadcastTest()
404 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeBroadcastTest()
|