Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
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 217 of file common_cudnn.h.


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