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

Namespaces

 dragon
 

Functions

 dragon::DEPLOY_CPU (DepthwiseConv2d)
 
 dragon::DEPLOY_CUDA (DepthwiseConv2d)
 
 dragon::DEPLOY_CPU (DepthwiseConv2dGradient)
 
 dragon::DEPLOY_CUDA (DepthwiseConv2dGradient)
 
 dragon::NumInputs (2, 3) .NumOutputs(1) .Inplace(
 
 dragon::NumInputs (3) .NumOutputs(2)
 
 dragon::REGISTER_GRADIENT (DepthwiseConv2d, GradientMaker)