Searched defs:num_ranks_per_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 60 const int num_ranks_per_node; member
88 static void PopulateMultiNodeParams(int* num_nodes, int* num_ranks_per_node) { in PopulateMultiNodeParams()
104 TestCase* MakeReductionTestCase(int num_nodes, int num_ranks_per_node, in MakeReductionTestCase()
165 TestCase* MakeGatherTestCase(int num_nodes, int num_ranks_per_node, in MakeGatherTestCase()
210 TestCase* MakeBroadcastTestCase(int num_nodes, int num_ranks_per_node, in MakeBroadcastTestCase()
303 const int num_ranks_per_node) { in RunMultiNodeAllReduceTest()
309 const int num_ranks_per_node) { in RunMultiNodeAllReduceTest()
363 const int num_ranks_per_node, in RunMultiNodeBroadcastTest()
421 static int GlobalRank(int num_ranks_per_node, int node, int local_rank) { in GlobalRank()
425 static BaseGPUDevice* GetDevice(int num_ranks_per_node, int node, in GetDevice()
[all …]
|