caffe2::cudnnTensorDescWrapper Class Reference

cudnnTensorDescWrapper is the placeholder that wraps around a cudnnTensorDescriptor_t, allowing us to do descriptor change as-needed during runtime. More...

#include <common_cudnn.h>

Public Member Functions
cudnnTensorDescriptor_t	Descriptor (const cudnnTensorFormat_t format, const cudnnDataType_t type, const vector< int > &dims, bool *changed)
template<typename T >
cudnnTensorDescriptor_t	Descriptor (const StorageOrder &order, const vector< int > &dims)

Detailed Description

cudnnTensorDescWrapper is the placeholder that wraps around a cudnnTensorDescriptor_t, allowing us to do descriptor change as-needed during runtime.

Definition at line 179 of file common_cudnn.h.

---

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

• caffe2/core/common_cudnn.h