|
| 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 |
|
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: