Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Namespaces | Functions
elu_op.cc File Reference
#include "utils/math_functions.h"
#include "utils/op_kernel.h"
#include "operators/activation/elu_op.h"

Namespaces

 dragon
 

Functions

 dragon::DEPLOY_CPU (Elu)
 
 dragon::DEPLOY_CUDA (Elu)
 
 dragon::DEPLOY_CPU (EluGradient)
 
 dragon::DEPLOY_CUDA (EluGradient)
 
 dragon::NumInputs (1) .NumOutputs(1) .Inplace(
 
 dragon::NumInputs (2) .NumOutputs(1) .Inplace(
 
 dragon::REGISTER_GRADIENT (Elu, InplaceGradientMaker)