Home
last modified time | relevance | path

Searched defs:BlockHostUntilDone (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc111 port::Status XlaInterpreterExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::interpreter::XlaInterpreterExecutor
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc63 Status TpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc271 port::Status HostExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc550 port::Status BlockHostUntilDone(Stream* stream) override { in BlockHostUntilDone() function in stream_executor::__anon00950a620111::CStreamExecutor
Dstream_executor_test.cc851 TEST_F(StreamExecutorTest, BlockHostUntilDone) { in TEST_F() argument
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc742 port::Status GpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc650 port::Status GpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc475 port::Status StreamExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::StreamExecutor
Dstream.cc4970 port::Status Stream::BlockHostUntilDone() { in BlockHostUntilDone() function in stream_executor::Stream