Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Static Public Member Functions | Static Public Attributes
caffe2::cudnnTypeWrapper< at::Half > Class Template Reference

Public Types

typedef const float ScalingParamType
 
typedef float BNParamType
 

Static Public Member Functions

static ScalingParamType * kOne ()
 
static ScalingParamType * kZero ()
 

Static Public Attributes

static const cudnnDataType_t type = CUDNN_DATA_HALF
 

Detailed Description

template<>
class caffe2::cudnnTypeWrapper< at::Half >

Definition at line 173 of file common_cudnn.h.


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