Home
last modified time | relevance | path

Searched refs:HostStream (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_stream.cc23 HostStream::HostStream() in HostStream() function in stream_executor::host::HostStream
28 HostStream::~HostStream() {} in ~HostStream()
30 bool HostStream::EnqueueTask(std::function<void()> task) { in EnqueueTask()
32 HostStream* stream; in EnqueueTask()
56 void HostStream::BlockUntilDone() { in BlockUntilDone()
Dhost_stream.h30 class HostStream : public internal::StreamInterface {
32 HostStream();
33 ~HostStream() override;
Dhost_gpu_executor.cc32 HostStream *AsHostStream(Stream *stream) { in AsHostStream()
34 return dynamic_cast<HostStream *>(stream->implementation()); in AsHostStream()
Dhost_gpu_executor.h199 return std::unique_ptr<internal::StreamInterface>(new HostStream()); in GetStreamImplementation()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc25 host::HostStream *AsExecutorStream(Stream *stream) { in AsExecutorStream()
27 return dynamic_cast<host::HostStream *>(stream->implementation()); in AsExecutorStream()
Dexecutor.h199 return std::unique_ptr<internal::StreamInterface>(new host::HostStream()); in GetStreamImplementation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc297 auto* host_stream = dynamic_cast<se::host::HostStream*>( in ExecuteAsyncOnStreamImpl()