Searched refs:kNcclAllReduceDone (Results 1 – 4 of 4) sorted by relevance
63 kNcclAllReduceDone, enumerator
52 case Thunk::kNcclAllReduceDone: in KindToString()
326 : Thunk(Thunk::kNcclAllReduceDone, thunk_info), start_thunk_(start_thunk) {} in NcclAllReduceDoneThunk()
83 case Thunk::Kind::kNcclAllReduceDone: in NeedsAsyncCommsStream()