Searched refs:GenerateCommunicatorKey (Results 1 – 7 of 7) sorted by relevance
31 string GenerateCommunicatorKey() override { in GenerateCommunicatorKey() function in tensorflow::NcclCommunicator32 return nccl_manager_.GenerateCommunicatorKey(); in GenerateCommunicatorKey()
320 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeAllReduceTest()371 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeBroadcastTest()637 NcclManager::instance()->GenerateCommunicatorKey(); in TEST()742 rank == 0 ? "" : NcclManager::instance()->GenerateCommunicatorKey(), in TYPED_TEST()896 string communicator_key = nodes[0].nccl_manager.GenerateCommunicatorKey(); in TYPED_TEST()938 communicator_key = nodes[0].nccl_manager.GenerateCommunicatorKey(); in TYPED_TEST()
64 string GenerateCommunicatorKey();
261 string NcclManager::GenerateCommunicatorKey() { in GenerateCommunicatorKey() function in tensorflow::NcclManager
418 virtual string GenerateCommunicatorKey() = 0;
97 string GenerateCommunicatorKey() override { return "mock-communicator-key"; } in GenerateCommunicatorKey() function in tensorflow::__anon4a737d780111::FakeNcclCommunicator
150 nccl_communicator_->GenerateCommunicatorKey(); in CompleteGroupLocal()