Searched refs:IncrementalBarrier (Results 1 – 4 of 4) sorted by relevance
28 explicit InternalIncrementalBarrier(IncrementalBarrier::DoneCallback callback) in InternalIncrementalBarrier()35 IncrementalBarrier::DoneCallback done_callback = in operator ()()42 IncrementalBarrier::BarrierCallback Inc() { in Inc()51 IncrementalBarrier::DoneCallback done_callback_;54 IncrementalBarrier::IncrementalBarrier(DoneCallback done_callback) in IncrementalBarrier() function in tensorflow::IncrementalBarrier58 IncrementalBarrier::~IncrementalBarrier() { (*internal_barrier_)(); } in ~IncrementalBarrier()60 IncrementalBarrier::BarrierCallback IncrementalBarrier::Inc() { in Inc()
55 IncrementalBarrier::DoneCallback done_callback = in TEST()57 IncrementalBarrier barrier(done_callback); in TEST()66 IncrementalBarrier::BarrierCallback bc1, bc2; in TEST()68 IncrementalBarrier::DoneCallback done_callback = in TEST()70 IncrementalBarrier barrier(done_callback); in TEST()77 IncrementalBarrier::BarrierCallback bc3 = barrier.Inc(); in TEST()96 IncrementalBarrier::DoneCallback done_callback = in TEST()98 IncrementalBarrier barrier(done_callback); in TEST()107 IncrementalBarrier::BarrierCallback bc = barrier.Inc(); in TEST()123 IncrementalBarrier barrier([] {}); in BM_FunctionInc()
59 class IncrementalBarrier {63 explicit IncrementalBarrier(DoneCallback callback);65 ~IncrementalBarrier();
418 IncrementalBarrier barrier(split_task_done_callback); in SplitInputTask()