Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
caffe2::CudaEventWrapper Struct Reference

Public Member Functions

 CudaEventWrapper (const DeviceOption &option)
 

Data Fields

cudaEvent_t cuda_event_
 
cudaStream_t cuda_stream_
 
int device_id_
 
std::atomic< int > status_
 
std::mutex mutex_recorded_
 
std::condition_variable cv_recorded_
 
std::string err_msg_
 

Detailed Description

Definition at line 9 of file event_gpu.cc.


The documentation for this struct was generated from the following file: