Searched refs:GpuEventHandle (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | gpu_driver.h | 91 static port::Status InitEvent(GpuContext* context, GpuEventHandle* result, 97 static port::Status DestroyEvent(GpuContext* context, GpuEventHandle* event); 383 GpuEventHandle event); 428 GpuEventHandle start, GpuEventHandle stop); 433 static port::Status RecordEvent(GpuContext* context, GpuEventHandle event, 440 GpuEventHandle event);
|
D | gpu_event.h | 49 GpuEventHandle gpu_event(); 56 GpuEventHandle gpu_event_;
|
D | gpu_timer.h | 74 GpuEventHandle start_event_; // Event recorded to indicate the "start" 76 GpuEventHandle stop_event_; // Event recorded to indicate the "stop"
|
D | gpu_stream.h | 64 GpuEventHandle* completed_event() { return &completed_event_; } in completed_event() 86 GpuEventHandle completed_event_ = nullptr;
|
D | gpu_types.h | 48 using GpuEventHandle = hipEvent_t; variable 66 using GpuEventHandle = CUevent;
|
D | gpu_event.cc | 44 GpuEventHandle GpuEvent::gpu_event() { return gpu_event_; } in gpu_event()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_driver.cc | 697 GpuEventHandle* event) { in DestroyEvent() 725 GpuEventHandle event, in RecordEvent() 747 GpuContext* context, GpuEventHandle event) { in QueryEvent() 761 GpuEventHandle start, in GetEventElapsedTime() 762 GpuEventHandle stop) { in GetEventElapsedTime() 784 GpuEventHandle event) { in WaitStreamOnEvent() 949 GpuEventHandle* event, in InitEvent()
|
D | rocm_gpu_executor.cc | 628 GpuEventHandle other_completed_event = *AsGpuStream(other)->completed_event(); in CreateStreamDependency()
|