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

Namespaces

 dragon
 

Macros

#define DETERMINE_RUNTIME_ARGS(X)
 

Functions

 dragon::DEPLOY_CPU (Concat)
 
 dragon::DEPLOY_CUDA (Concat)
 
 dragon::DEPLOY_CPU (ConcatGradient)
 
 dragon::DEPLOY_CUDA (ConcatGradient)
 
 dragon::NumInputs (1, INT_MAX) .NumOutputs(1
 
 dragon::NumInputs (2, INT_MAX) .NumOutputs(1)
 
 dragon::REGISTER_GRADIENT (Concat, SimpleGradientMaker)
 

Macro Definition Documentation

◆ DETERMINE_RUNTIME_ARGS

#define DETERMINE_RUNTIME_ARGS (   X)
Value:
axis_ = OpArg<int64_t>("axis", 0); \
axis_ = axis_ < 0 ? axis_ + X.ndim() : axis_; \
CHECK(axis_ >= 0 && axis_ < X.ndim()) \
<< "\nExcepted the axis in [-" << X.ndim() \
<< ", " << X.ndim() << "), got " \
<< OpArg<int64_t>("axis", 0) << ".";