Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::CuDNNState Class Reference

Public Member Functions

 CuDNNState (size_t gpu_id)
 
cudnnHandle_t & cudnn_handle ()
 
CuDNNWorkspaceworkspace ()
 
template<typename F >
void execute (cudaStream_t stream, F &&f)
 

Detailed Description

Definition at line 49 of file cudnn_wrappers.h.


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