Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
sqrt_op.h
Go to the documentation of this file.
1 
13 #ifndef DRAGON_OPERATORS_ARITHMETIC_SQRT_OP_H_
14 #define DRAGON_OPERATORS_ARITHMETIC_SQRT_OP_H_
15 
16 #include "core/operator.h"
17 
18 namespace dragon {
19 
20 template <class Context>
21 class SqrtOp 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 SqrtGradientOp 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_SQRT_OP_H_
void RunOnDevice() override
Implement the detailed execution.
Definition: sqrt_op.cc:32
void RunOnDevice() override
Implement the detailed execution.
Definition: sqrt_op.cc:14
Definition: sqrt_op.h:31
void RunImpl()
Definition: sqrt_op.cc:7
USE_OPERATOR_FUNCTIONS
Definition: sqrt_op.h:24
Definition: sqrt_op.h:21
Definition: operator.h:149
void RunImpl()
Definition: sqrt_op.cc:23
USE_OPERATOR_FUNCTIONS
Definition: sqrt_op.h:34
SIMPLE_CTOR_DTOR(SqrtOp)
SIMPLE_CTOR_DTOR(SqrtGradientOp)
Definition: common.h:41