HomeSort by relevance Sort by last modified time
    Searched refs:HostStream (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/stream_executor/host/
host_stream.cc 16 // Class method definitions for HostStream, the Stream implementation for
24 HostStream::HostStream()
29 HostStream::~HostStream() {}
31 bool HostStream::EnqueueTask(std::function<void()> task) {
47 void HostStream::BlockUntilDone() {
host_stream.h 31 class HostStream : public internal::StreamInterface {
33 HostStream();
34 ~HostStream() override;
host_gpu_executor.cc 35 HostStream *AsHostStream(Stream *stream) {
37 return dynamic_cast<HostStream *>(stream->implementation());
71 // Enqueue the [asynchronous] memcpy on the stream (HostStream) associated
82 // Enqueue the [asynchronous] memcpy on the stream (HostStream) associated
96 // the nature of the HostExecutor) memcpy on the stream (HostStream)
106 // Enqueue the [asynchronous] memzero on the stream (HostStream) associated
116 // Enqueue the [asynchronous] memzero on the stream (HostStream) associated
126 // Enqueue the [asynchronous] memzero on the stream (HostStream) associated
host_gpu_executor.h 199 return std::unique_ptr<internal::StreamInterface>(new HostStream());
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.cc 26 host::HostStream *AsExecutorStream(Stream *stream) {
28 return dynamic_cast<host::HostStream *>(stream->implementation());
executor.h 199 return std::unique_ptr<internal::StreamInterface>(new host::HostStream());
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.cc 325 auto* host_stream = dynamic_cast<perftools::gputools::host::HostStream*>(

Completed in 222 milliseconds