Searched refs:SignalMultiNodeReady (Results 1 – 4 of 4) sorted by relevance
192 void SignalMultiNodeReady(const string& collective_key);
189 nccl_manager_.SignalMultiNodeReady(nccl_collective_key); in Enqueue()
352 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeAllReduceTest()410 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeBroadcastTest()892 nodes[node].nccl_manager.SignalMultiNodeReady(collective_key); in TYPED_TEST()
470 void NcclManager::SignalMultiNodeReady(const string& collective_key) { in SignalMultiNodeReady() function in tensorflow::NcclManager