Searched refs:all_reduce_start_thunks_ (Results 1 – 2 of 2) sorted by relevance
768 all_reduce_start_thunks_; variable
2845 all_reduce_start_thunks_, op, thunk.get())); in EmitNcclThunk()2852 MaybeAddAllReduceStartThunkToMap(all_reduce_start_thunks_, op, nullptr)); in EmitNcclThunk()2898 auto it = all_reduce_start_thunks_.find(start_op); in EmitAllReduceDone()2899 TF_RET_CHECK(it != all_reduce_start_thunks_.end()) in EmitAllReduceDone()2908 all_reduce_start_thunks_.erase(it); in EmitAllReduceDone()