Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
Data Structures
Data Structure Index
Class Hierarchy
Data Fields
caffe2
CuDNNState
Public Member Functions
caffe2::CuDNNState Class Reference
Public Member Functions
CuDNNState
(size_t gpu_id)
cudnnHandle_t &
cudnn_handle
()
CuDNNWorkspace
&
workspace
()
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:
caffe2/core/
cudnn_wrappers.h
Generated on Thu Apr 19 2018 13:03:58 for Caffe2 - C++ API by
1.8.11