Searched refs:ThenDoHostCallback (Results 1 – 8 of 8) sorted by relevance
28 return stream->ThenDoHostCallback([this]() { this->StartNow(); }).ok(); in Start()32 return stream->ThenDoHostCallback([this]() { this->StopNow(); }).ok(); in Stop()
50 if (!stream->ThenDoHostCallback([buffers_raw] { delete[] buffers_raw; }) in ExecuteOnStream()
92 .ThenDoHostCallback([buffer]() { (*buffer)->Done(); }); in ExecuteOnStream()
189 host_to_device_stream_->ThenDoHostCallback([ref]() { ref.Unref(); }); in CopyCPUTensorToDevice()192 host_to_device_stream_->ThenDoHostCallback([ref, done]() { in CopyCPUTensorToDevice()
1017 stream->ThenDoHostCallback(allreduce_launch_callback); in ComputeAsync()1111 stream->ThenDoHostCallback(allgather_launch_callback); in ComputeAsync()
1071 stream->ThenDoHostCallback(allreduce_launch_callback); in ComputeAsync()1187 stream->ThenDoHostCallback(allgather_launch_callback); in ComputeAsync()
1929 Stream &ThenDoHostCallback(std::function<void()> callback);
5116 Stream &Stream::ThenDoHostCallback(std::function<void()> callback) { in ThenDoHostCallback() function in stream_executor::Stream5254 return ThenDoHostCallback([stream_executor, bound_task]() { in ThenEnqueueOnBackgroundThread()