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

Namespaces

 dragon
 

Macros

#define WORKSPACE_LIMIT_BYTES   1024 * 1024 * 1024
 

Functions

 dragon::DEPLOY_CUDNN (Conv2d)
 
 dragon::DEPLOY_CUDNN (Conv2dGradient)
 

Macro Definition Documentation

◆ WORKSPACE_LIMIT_BYTES

#define WORKSPACE_LIMIT_BYTES   1024 * 1024 * 1024