Home
last modified time | relevance | path

Searched refs:MakeBroadcastTestCase (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc210 TestCase* MakeBroadcastTestCase(int num_nodes, int num_ranks_per_node, in MakeBroadcastTestCase() function in tensorflow::NcclManagerTest
372 std::unique_ptr<TestCase> test_case(this->MakeBroadcastTestCase( in RunMultiNodeBroadcastTest()
784 this->MakeBroadcastTestCase(/*num_nodes=*/1, num_ranks, in TYPED_TEST()
810 this->MakeBroadcastTestCase(/*num_nodes=*/1, num_ranks, in TYPED_TEST()
838 this->MakeBroadcastTestCase(/*num_nodes=*/1, num_ranks, in TYPED_TEST()