Home
last modified time | relevance | path

Searched refs:EnqueueTask (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc71 AsHostStream(stream)->EnqueueTask( in Memcpy()
81 AsHostStream(stream)->EnqueueTask( in Memcpy()
95 AsHostStream(stream)->EnqueueTask( in MemcpyDeviceToDevice()
105 AsHostStream(stream)->EnqueueTask( in MemZero()
115 AsHostStream(stream)->EnqueueTask( in Memset()
125 AsHostStream(stream)->EnqueueTask( in Memset32()
152 AsHostStream(stream)->EnqueueTask([callback]() { in HostCallback()
166 AsHostStream(dependent)->EnqueueTask( in CreateStreamDependency()
Dhost_stream.h35 bool EnqueueTask(std::function<void()> task);
Dhost_stream.cc30 bool HostStream::EnqueueTask(std::function<void()> task) { in EnqueueTask() function in stream_executor::host::HostStream
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc53 AsExecutorStream(stream)->EnqueueTask([this, host_dst, dev_src, size]() { in Memcpy()
62 AsExecutorStream(stream)->EnqueueTask([this, dev_dst, host_src, size]() { in Memcpy()
83 AsExecutorStream(stream)->EnqueueTask([callback]() { in HostCallback()
94 AsExecutorStream(dependent)->EnqueueTask( in CreateStreamDependency()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc339 host_stream->EnqueueTask( in ExecuteAsyncOnStreamImpl()