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

Public Member Functions

 HardSigmoidGradientFunctor (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::HardSigmoidGradientFunctor< Context >

Definition at line 23 of file hard_sigmoid_op.h.


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