Home
last modified time | relevance | path

Searched refs:GpuEvent (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_event.cc25 GpuEvent::GpuEvent(GpuExecutor* parent) in GpuEvent() function in stream_executor::gpu::GpuEvent
28 GpuEvent::~GpuEvent() {} in ~GpuEvent()
30 port::Status GpuEvent::Init() { in Init()
35 port::Status GpuEvent::Destroy() { in Destroy()
39 port::Status GpuEvent::Record(GpuStream* stream) { in Record()
44 GpuEventHandle GpuEvent::gpu_event() { return gpu_event_; } in gpu_event()
Dgpu_event.h29 class GpuEvent : public internal::EventInterface {
31 explicit GpuEvent(GpuExecutor* parent);
33 ~GpuEvent() override;
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_event.h24 using CUDAEvent = gpu::GpuEvent;
Dcuda_event.cc25 Event::Status GpuEvent::PollForStatus() { in PollForStatus()
Dcuda_gpu_executor.cc86 static GpuEvent* AsGpuEvent(Event* event) { in AsGpuEvent()
88 return static_cast<GpuEvent*>(event->implementation()); in AsGpuEvent()
941 return std::unique_ptr<internal::EventInterface>(new GpuEvent(this)); in CreateEventImplementation()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_event.cc24 Event::Status GpuEvent::PollForStatus() { in PollForStatus()
Drocm_gpu_executor.cc63 static GpuEvent* AsGpuEvent(Event* event) { in AsGpuEvent()
65 return static_cast<GpuEvent*>(event->implementation()); in AsGpuEvent()
843 return std::unique_ptr<internal::EventInterface>(new GpuEvent(this)); in CreateEventImplementation()