Searched refs:GpuStream (Results 1 – 9 of 9) sorted by relevance
25 bool GpuStream::Init() { in Init()34 void GpuStream::Destroy() { in Destroy()46 bool GpuStream::IsIdle() const { in IsIdle()50 GpuStream* AsGpuStream(Stream* stream) { in AsGpuStream()52 return static_cast<GpuStream*>(stream->implementation()); in AsGpuStream()
35 class GpuStream : public internal::StreamInterface {37 explicit GpuStream(GpuExecutor* parent) in GpuStream() function41 ~GpuStream() override {} in ~GpuStream()88 GpuStream* AsGpuStream(Stream* stream);
30 class GpuStream; variable54 bool Start(GpuStream* stream);57 bool Stop(GpuStream* stream);
74 bool GpuTimer::Start(GpuStream* stream) { in Start()83 bool GpuTimer::Stop(GpuStream* stream) { in Stop()
43 port::Status Record(GpuStream* stream);
39 port::Status GpuEvent::Record(GpuStream* stream) { in Record()
27 using CUDAStream = gpu::GpuStream;
732 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream()951 return std::unique_ptr<internal::StreamInterface>(new GpuStream(this)); in GetStreamImplementation()
613 GpuStream* rocm_stream = AsGpuStream(stream); in DeallocateStream()853 return std::unique_ptr<internal::StreamInterface>(new GpuStream(this)); in GetStreamImplementation()