Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
operators
sigmoid_op.h
1
#ifndef CAFFE2_OPERATORS_SIGMOID_OP_H_
2
#define CAFFE2_OPERATORS_SIGMOID_OP_H_
3
4
#include <vector>
5
6
#include "caffe2/operators/elementwise_ops.h"
7
8
namespace
caffe2
{
9
10
template
<
class
Context>
11
struct
SigmoidFunctor
{
12
template
<
typename
T>
13
bool
operator()(
const
int
N,
const
T
* X,
T
* Y, Context* context)
const
;
14
};
15
16
template
<
class
Context>
17
struct
SigmoidGradientFunctor
{
18
template
<
typename
T>
19
bool
Forward(
20
const
std::vector<int>& Y_dims,
21
const
std::vector<int>& dY_dims,
22
const
T
* Y,
23
const
T
* dY,
24
T
* dX,
25
Context* context)
const
;
26
};
27
28
}
// namespace caffe2
29
30
#endif // CAFFE2_OPERATORS_SIGMOID_OP_H_
caffe2::SigmoidGradientFunctor
Definition:
sigmoid_op.h:17
T
Definition:
dataloader.cpp:482
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
blob.h:13
caffe2::SigmoidFunctor
Definition:
sigmoid_op.h:11
Generated on Thu Mar 21 2019 13:06:19 for Caffe2 - C++ API by
1.8.11