Searched defs:done_callback (Results 1 – 11 of 11) sorted by relevance
35 std::function<void(const Status& status, const EnqueueResponse& response)> in RemoteExecuteNode()
140 auto done_callback = [&nccl_done, &nccl_status](const Status& s) { in Run() local
94 DoneCallback done_callback; // must be run outside mu_ member
160 DoneCallback done_callback; // must be run outside mu_ member
242 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput()289 AsyncOpKernel::DoneCallback done_callback; member982 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute() local
108 DoneCallback done_callback; member
618 auto done_callback = [collective, p_idx, nccl_result]() { in LoopKernelLaunches() local
43 std::function<void()> done_callback) in TestDelegate()
604 def _wait_cancel(self, event, done_callback): argument
149 def done_callback(): function
1070 auto done_callback = base::Bind( in TEST_F() local