Searched defs:done_callback (Results 1 – 14 of 14) sorted by relevance
35 IncrementalBarrier::DoneCallback done_callback = in operator ()() local54 IncrementalBarrier::IncrementalBarrier(DoneCallback done_callback) in IncrementalBarrier()
55 IncrementalBarrier::DoneCallback done_callback = in TEST() local68 IncrementalBarrier::DoneCallback done_callback = in TEST() local96 IncrementalBarrier::DoneCallback done_callback = in TEST() local
81 StatusCallback done_callback; in Run() local
94 DoneCallback done_callback; // must be run outside mu_ member
156 DoneCallback done_callback; // must be run outside mu_ member
651 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute() local
125 DoneCallback done_callback; member
834 auto done_callback = [collective, p_idx, nccl_result]() { in LoopKernelLaunches() local
72 AsyncOpKernel::DoneCallback done_callback; member
132 int split_index, AsyncOpKernel::DoneCallback done_callback) { in CreateSplitTask()156 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput()
42 std::function<void()> done_callback) in TestDelegate()
672 def _wait_cancel(self, event, done_callback): argument
149 def done_callback(): function
1070 auto done_callback = base::Bind( in TEST_F() local