Caffe2 - C++ API
A deep learning, cross platform ML framework
softsign_op.h
1 #ifndef CAFFE2_OPERATORS_SOFTSIGN_OP_H_
2 #define CAFFE2_OPERATORS_SOFTSIGN_OP_H_
3 
4 #include <vector>
5 
6 #include "caffe2/operators/elementwise_ops.h"
7 #include "caffe2/utils/math.h"
8 
9 namespace caffe2 {
10 
11 template <class Context>
13  template <typename T>
14  bool operator()(const int N, const T* X, T* Y, Context* context) const;
15 };
16 
17 template <class Context>
19  template <typename T>
20  bool Forward(
21  const std::vector<int>& X_dims,
22  const std::vector<int>& dY_dims,
23  const T* X,
24  const T* dY,
25  T* dX,
26  Context* context) const;
27 };
28 
29 } // namespace caffe2
30 
31 #endif // CAFFE2_OPERATORS_SOFTSIGN_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13