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

Public Member Functions

 HardSigmoidFunctor (OperatorBase &op)
 
template<typename T >
bool operator() (const int N, const T *X, T *Y, Context *context) const
 
template<>
bool operator() (const int N, const T *X, T *Y, CPUContext *) const
 

Data Fields

const float alpha
 
const float beta
 

Detailed Description

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

Definition at line 11 of file hard_sigmoid_op.h.


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