Searched refs:callback_stream_ (Results 1 – 2 of 2) sorted by relevance
110 se::Stream* callback_stream() const { return callback_stream_.get(); } in callback_stream()151 if (callback_stream_.get() != stream) { in ThenRelease()152 callback_stream_->ThenWaitFor(stream); in ThenRelease()154 ThenExecuteOnCallbackThread(callback_stream_.get(), in ThenRelease()197 std::unique_ptr<se::Stream> callback_stream_; variable
44 callback_stream_ = absl::make_unique<se::Stream>(executor); in LocalDeviceState()47 callback_stream_->Init(); in LocalDeviceState()81 status.Update(callback_stream_->BlockHostUntilDone()); in SynchronizeAllActivity()