Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.cc129 port::Status XlaInterpreterExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::interpreter::XlaInterpreterExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.cc266 port::Status HostExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::host::HostExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor.cc63 Status TpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in tensorflow::tpu::TpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc658 port::Status GpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc537 port::Status BlockHostUntilDone(Stream* stream) override { in BlockHostUntilDone() function in stream_executor::__anon702966b10111::CStreamExecutor
H A Dstream_executor_test.cc675 TEST_F(StreamExecutorTest, BlockHostUntilDone) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc790 port::Status GpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc485 port::Status StreamExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone() function in stream_executor::StreamExecutor
H A Dstream.cc2479 port::Status Stream::BlockHostUntilDone() { in BlockHostUntilDone() function in stream_executor::Stream