Searched refs:RunMultiNodeBroadcastTest (Results 1 – 1 of 1) sorted by relevance
362 void RunMultiNodeBroadcastTest(const int num_nodes, in RunMultiNodeBroadcastTest() function in tensorflow::NcclManagerTest607 this->RunMultiNodeBroadcastTest(/*num_nodes=*/1, in TYPED_TEST()615 this->RunMultiNodeBroadcastTest(/*num_nodes=*/1, in TYPED_TEST()627 this->RunMultiNodeBroadcastTest(/*num_nodes=*/1, num_ranks, in TYPED_TEST()678 this->RunMultiNodeBroadcastTest(num_nodes, num_ranks_per_node, in TYPED_TEST()