Searched refs:CUDAEvent (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
H A Dcuda_event.h28 // CUDAEvent wraps a CUevent in the platform-independent EventInterface
30 class CUDAEvent : public internal::EventInterface { class in namespace:perftools::gputools::cuda
32 explicit CUDAEvent(CUDAExecutor* parent);
34 ~CUDAEvent() override;
H A Dcuda_event.cc26 CUDAEvent::CUDAEvent(CUDAExecutor* parent) function in class:perftools::gputools::cuda::CUDAEvent
29 CUDAEvent::~CUDAEvent() {}
31 port::Status CUDAEvent::Init() {
36 port::Status CUDAEvent::Destroy() {
40 port::Status CUDAEvent::Record(CUDAStream* stream) {
45 Event::Status CUDAEvent::PollForStatus() {
66 const CUevent& CUDAEvent::cuda_event() {
H A Dcuda_gpu_executor.cc83 static CUDAEvent *AsCUDAEvent(Event *event) {
85 return static_cast<CUDAEvent *>(event->implementation());
830 return std::unique_ptr<internal::EventInterface>(new CUDAEvent(this));

Completed in 114 milliseconds