Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc47 void WaitAndLogIfStuck(tensorflow::BlockingCounter* counter, in WaitAndLogIfStuck() function
181 WaitAndLogIfStuck(&all_arrived_, [&] { in SubmitParticipant()
282 WaitAndLogIfStuck(final_sync.get(), [&] { in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h123 void WaitAndLogIfStuck(tensorflow::BlockingCounter* counter, in WaitAndLogIfStuck() function
233 xla::WaitAndLogIfStuck(blocking_counter.get(), [&] { in SubmitParticipant()
290 WaitAndLogIfStuck(&all_participants_present_, [&] { in SubmitParticipant()