Searched refs:ThenDoHostCallback (Results 1 – 13 of 13) sorted by relevance
28 return stream->ThenDoHostCallback([this]() { this->StartNow(); }).ok(); in Start()32 return stream->ThenDoHostCallback([this]() { this->StopNow(); }).ok(); in Stop()
37 stream.ThenDoHostCallback([i, &mu, &expected, &ok]() { in TEST()
197 host_to_device_stream_->ThenDoHostCallback([ref]() { ref.Unref(); }); in CopyCPUTensorToDevice()200 host_to_device_stream_->ThenDoHostCallback([ref, done]() { in CopyCPUTensorToDevice()282 status_or_new_stream.ValueOrDie()->ThenDoHostCallback( in CopyDeviceTensorToCPU()301 stream->ThenDoHostCallback(std::move(func)); in ThenExecute()
282 dst_device_to_device_stream->ThenDoHostCallback( in TpuDeviceToDeviceCopy()
235 stream->ThenDoHostCallback([raw_input_runtime, padded_data, in UpdateDynamicInputs()273 stream->ThenDoHostCallback([padded_data]() { return Status::OK(); }); in UpdateDynamicInputs()374 deregister_stream->ThenDoHostCallback([=]() { in UnregisterCancellation()
98 .ThenDoHostCallback([&buffer]() { buffer->Done(); }); in ExecuteOnStream()
219 main_stream->ThenDoHostCallback(std::move(fn)); in ExecuteThunks()
55 stream->ThenDoHostCallback([element_pointers{std::move(element_pointers)}]() { in WriteSingleTupleIndexTable()
270 stream->ThenDoHostCallback([profile, device_description]() { in ExecuteWrapperAfterExecution()
101 stream->ThenDoHostCallback([this, callback]() mutable { in ThenExecuteOnCallbackThread()
176 stream->ThenDoHostCallback([metadata_literal]() { return Status::OK(); }); in UpdateMetadata()
1983 Stream &ThenDoHostCallback(std::function<void()> callback);
4825 Stream &Stream::ThenDoHostCallback(std::function<void()> callback) { in ThenDoHostCallback() function in stream_executor::Stream4965 return ThenDoHostCallback([stream_executor, bound_task]() { in ThenEnqueueOnBackgroundThread()