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

Namespaces

 dragon
 

Macros

#define DETERMINE_RUNTIME_ARGS(X)
 

Functions

 dragon::DEPLOY_CPU (Softmax)
 
 dragon::DEPLOY_CUDA (Softmax)
 
 dragon::DEPLOY_CPU (SoftmaxGradient)
 
 dragon::DEPLOY_CUDA (SoftmaxGradient)
 
 dragon::NumInputs (1) .NumOutputs(1) .Inplace(
 
 dragon::NumInputs (2) .NumOutputs(1) .Inplace(
 
 dragon::REGISTER_GRADIENT (Softmax, InplaceGradientMaker)
 

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 axis in [-" << X.ndim() << ", " << X.ndim() \
<< "), got " << OpArg<int64_t>("axis", 0) << ".";