Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Namespaces | Functions
batch_norm.cc File Reference
#include "core/workspace.h"
#include "utils/filler.h"
#include "utils/op_kernel.h"
#include "utils/math_functions.h"
#include "operators/norm/batch_norm_op.h"

Namespaces

 dragon
 

Functions

 dragon::DEPLOY_CPU (BatchNorm)
 
 dragon::DEPLOY_CUDA (BatchNorm)
 
 dragon::DEPLOY_CPU (BatchNormGradient)
 
 dragon::DEPLOY_CUDA (BatchNormGradient)
 
 dragon::NumInputs (5) .NumOutputs(1)
 
 dragon::REGISTER_GRADIENT (BatchNorm, GradientMaker)