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 354 of file common_cudnn.h.

---

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

• caffe2/core/common_cudnn.h