Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
caffe2::EluGradientFunctor< Context > Struct Template Reference

Public Member Functions

 EluGradientFunctor (OperatorBase &op)
 
template<typename T >
bool Forward (const std::vector< int > &Y_dims, const std::vector< int > &dY_dims, const T *Y, const T *dY, T *dX, Context *context) const
 
template<>
bool Forward (const std::vector< int > &Y_dims, const std::vector< int > &, const T *Y, const T *dY, T *dX, CPUContext *) const
 

Data Fields

const float alpha
 

Detailed Description

template<class Context>
struct caffe2::EluGradientFunctor< Context >

Definition at line 22 of file elu_op.h.


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