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

Public Member Functions

 Stream (const DeviceOption &device_option)
 
void wait (Event *event) const
 

Data Fields

int gpu_id_ {-1}
 
cudaStream_t stream_ {nullptr}
 

Detailed Description

Definition at line 79 of file net_gpu.cc.


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