Searched refs:BlockUntilDone (Results 1 – 7 of 7) sorted by relevance
57 AsExecutorStream(stream)->BlockUntilDone(); in Memcpy()66 AsExecutorStream(stream)->BlockUntilDone(); in Memcpy()97 AsExecutorStream(dependent)->BlockUntilDone(); in CreateStreamDependency()112 AsExecutorStream(stream)->BlockUntilDone(); in BlockHostUntilDone()
44 void BlockUntilDone();
81 void HostStream::BlockUntilDone() { in BlockUntilDone() function in stream_executor::host::HostStream
272 AsHostStream(stream)->BlockUntilDone(); in BlockHostUntilDone()
46 Status BlockUntilDone() const;
40 Status AsyncExecution::BlockUntilDone() const { in BlockUntilDone() function in xla::AsyncExecution
906 TF_RETURN_IF_ERROR(execution->BlockUntilDone()); in WaitForExecution()