Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
exp_op.h
Go to the documentation of this file.
1 
13 #ifndef DRAGON_OPERATORS_ARITHMETIC_EXP_OP_H_
14 #define DRAGON_OPERATORS_ARITHMETIC_EXP_OP_H_
15 
16 #include "core/operator.h"
17 
18 namespace dragon {
19 
20 template <class Context>
21 class ExpOp final : public Operator<Context> {
22  public:
25 
26  void RunOnDevice() override;
27  template <typename T> void RunImpl();
28 };
29 
30 template <class Context>
31 class ExpGradientOp final : public Operator<Context> {
32  public:
35 
36  void RunOnDevice() override;
37  template <typename T> void RunImpl();
38 };
39 
40 } // namespace dragon
41 
42 #endif // DRAGON_OPERATORS_ARITHMETIC_EXP_OP_H_
Definition: exp_op.h:31
void RunOnDevice() override
Implement the detailed execution.
Definition: exp_op.cc:15
SIMPLE_CTOR_DTOR(ExpOp)
void RunImpl()
Definition: exp_op.cc:8
SIMPLE_CTOR_DTOR(ExpGradientOp)
USE_OPERATOR_FUNCTIONS
Definition: exp_op.h:24
Definition: operator.h:149
Definition: exp_op.h:21
USE_OPERATOR_FUNCTIONS
Definition: exp_op.h:34
void RunOnDevice() override
Implement the detailed execution.
Definition: exp_op.cc:32
Definition: common.h:41
void RunImpl()
Definition: exp_op.cc:24