mxnet
Macros
Argument indices
Collaboration diagram for Argument indices:

Macros

#define MKLDNN_ARG_SRC_0   1
 
#define MKLDNN_ARG_SRC   MKLDNN_ARG_SRC_0
 
#define MKLDNN_ARG_SRC_LAYER   MKLDNN_ARG_SRC_0
 
#define MKLDNN_ARG_FROM   MKLDNN_ARG_SRC_0
 
#define MKLDNN_ARG_SRC_1   2
 
#define MKLDNN_ARG_SRC_ITER   MKLDNN_ARG_SRC_1
 
#define MKLDNN_ARG_SRC_2   3
 
#define MKLDNN_ARG_SRC_ITER_C   MKLDNN_ARG_SRC_2
 
#define MKLDNN_ARG_DST_0   17
 
#define MKLDNN_ARG_DST   MKLDNN_ARG_DST_0
 
#define MKLDNN_ARG_TO   MKLDNN_ARG_DST_0
 
#define MKLDNN_ARG_DST_LAYER   MKLDNN_ARG_DST_0
 
#define MKLDNN_ARG_DST_1   18
 
#define MKLDNN_ARG_DST_ITER   MKLDNN_ARG_DST_1
 
#define MKLDNN_ARG_DST_2   19
 
#define MKLDNN_ARG_DST_ITER_C   MKLDNN_ARG_DST_2
 
#define MKLDNN_ARG_WEIGHTS_0   33
 
#define MKLDNN_ARG_WEIGHTS   MKLDNN_ARG_WEIGHTS_0
 
#define MKLDNN_ARG_SCALE_SHIFT   MKLDNN_ARG_WEIGHTS_0
 
#define MKLDNN_ARG_WEIGHTS_LAYER   MKLDNN_ARG_WEIGHTS_0
 
#define MKLDNN_ARG_WEIGHTS_1   34
 
#define MKLDNN_ARG_WEIGHTS_ITER   MKLDNN_ARG_WEIGHTS_1
 
#define MKLDNN_ARG_BIAS   41
 
#define MKLDNN_ARG_MEAN   49
 
#define MKLDNN_ARG_VARIANCE   50
 
#define MKLDNN_ARG_WORKSPACE   64
 
#define MKLDNN_ARG_SCRATCHPAD   80
 
#define MKLDNN_ARG_DIFF_SRC_0   129
 
#define MKLDNN_ARG_DIFF_SRC   MKLDNN_ARG_DIFF_SRC_0
 
#define MKLDNN_ARG_DIFF_SRC_LAYER   MKLDNN_ARG_DIFF_SRC_0
 
#define MKLDNN_ARG_DIFF_SRC_1   130
 
#define MKLDNN_ARG_DIFF_SRC_ITER   MKLDNN_ARG_DIFF_SRC_1
 
#define MKLDNN_ARG_DIFF_SRC_2   131
 
#define MKLDNN_ARG_DIFF_SRC_ITER_C   MKLDNN_ARG_DIFF_SRC_2
 
#define MKLDNN_ARG_DIFF_DST_0   145
 
#define MKLDNN_ARG_DIFF_DST   MKLDNN_ARG_DIFF_DST_0
 
#define MKLDNN_ARG_DIFF_DST_LAYER   MKLDNN_ARG_DIFF_DST_0
 
#define MKLDNN_ARG_DIFF_DST_1   146
 
#define MKLDNN_ARG_DIFF_DST_ITER   MKLDNN_ARG_DIFF_DST_1
 
#define MKLDNN_ARG_DIFF_DST_2   147
 
#define MKLDNN_ARG_DIFF_DST_ITER_C   MKLDNN_ARG_DIFF_DST_2
 
#define MKLDNN_ARG_DIFF_WEIGHTS_0   161
 
#define MKLDNN_ARG_DIFF_WEIGHTS   MKLDNN_ARG_DIFF_WEIGHTS_0
 
#define MKLDNN_ARG_DIFF_SCALE_SHIFT   MKLDNN_ARG_DIFF_WEIGHTS_0
 
#define MKLDNN_ARG_DIFF_WEIGHTS_LAYER   MKLDNN_ARG_DIFF_WEIGHTS_0
 
#define MKLDNN_ARG_DIFF_WEIGHTS_1   162
 
#define MKLDNN_ARG_DIFF_WEIGHTS_ITER   MKLDNN_ARG_DIFF_WEIGHTS_1
 
#define MKLDNN_ARG_DIFF_BIAS   169
 
#define MKLDNN_ARG_MULTIPLE_SRC   1024
 
#define MKLDNN_ARG_MULTIPLE_DST   2048
 

Detailed Description

Macro Definition Documentation

#define MKLDNN_ARG_BIAS   41
#define MKLDNN_ARG_DIFF_BIAS   169
#define MKLDNN_ARG_DIFF_DST   MKLDNN_ARG_DIFF_DST_0
#define MKLDNN_ARG_DIFF_DST_0   145
#define MKLDNN_ARG_DIFF_DST_1   146
#define MKLDNN_ARG_DIFF_DST_2   147
#define MKLDNN_ARG_DIFF_DST_ITER   MKLDNN_ARG_DIFF_DST_1
#define MKLDNN_ARG_DIFF_DST_ITER_C   MKLDNN_ARG_DIFF_DST_2
#define MKLDNN_ARG_DIFF_DST_LAYER   MKLDNN_ARG_DIFF_DST_0
#define MKLDNN_ARG_DIFF_SCALE_SHIFT   MKLDNN_ARG_DIFF_WEIGHTS_0
#define MKLDNN_ARG_DIFF_SRC   MKLDNN_ARG_DIFF_SRC_0
#define MKLDNN_ARG_DIFF_SRC_0   129
#define MKLDNN_ARG_DIFF_SRC_1   130
#define MKLDNN_ARG_DIFF_SRC_2   131
#define MKLDNN_ARG_DIFF_SRC_ITER   MKLDNN_ARG_DIFF_SRC_1
#define MKLDNN_ARG_DIFF_SRC_ITER_C   MKLDNN_ARG_DIFF_SRC_2
#define MKLDNN_ARG_DIFF_SRC_LAYER   MKLDNN_ARG_DIFF_SRC_0
#define MKLDNN_ARG_DIFF_WEIGHTS   MKLDNN_ARG_DIFF_WEIGHTS_0
#define MKLDNN_ARG_DIFF_WEIGHTS_0   161
#define MKLDNN_ARG_DIFF_WEIGHTS_1   162
#define MKLDNN_ARG_DIFF_WEIGHTS_ITER   MKLDNN_ARG_DIFF_WEIGHTS_1
#define MKLDNN_ARG_DIFF_WEIGHTS_LAYER   MKLDNN_ARG_DIFF_WEIGHTS_0
#define MKLDNN_ARG_DST   MKLDNN_ARG_DST_0
#define MKLDNN_ARG_DST_0   17
#define MKLDNN_ARG_DST_1   18
#define MKLDNN_ARG_DST_2   19
#define MKLDNN_ARG_DST_ITER   MKLDNN_ARG_DST_1
#define MKLDNN_ARG_DST_ITER_C   MKLDNN_ARG_DST_2
#define MKLDNN_ARG_DST_LAYER   MKLDNN_ARG_DST_0
#define MKLDNN_ARG_FROM   MKLDNN_ARG_SRC_0
#define MKLDNN_ARG_MEAN   49
#define MKLDNN_ARG_MULTIPLE_DST   2048
#define MKLDNN_ARG_MULTIPLE_SRC   1024
#define MKLDNN_ARG_SCALE_SHIFT   MKLDNN_ARG_WEIGHTS_0
#define MKLDNN_ARG_SCRATCHPAD   80
#define MKLDNN_ARG_SRC   MKLDNN_ARG_SRC_0
#define MKLDNN_ARG_SRC_0   1
#define MKLDNN_ARG_SRC_1   2
#define MKLDNN_ARG_SRC_2   3
#define MKLDNN_ARG_SRC_ITER   MKLDNN_ARG_SRC_1
#define MKLDNN_ARG_SRC_ITER_C   MKLDNN_ARG_SRC_2
#define MKLDNN_ARG_SRC_LAYER   MKLDNN_ARG_SRC_0
#define MKLDNN_ARG_TO   MKLDNN_ARG_DST_0
#define MKLDNN_ARG_VARIANCE   50
#define MKLDNN_ARG_WEIGHTS   MKLDNN_ARG_WEIGHTS_0
#define MKLDNN_ARG_WEIGHTS_0   33
#define MKLDNN_ARG_WEIGHTS_1   34
#define MKLDNN_ARG_WEIGHTS_ITER   MKLDNN_ARG_WEIGHTS_1
#define MKLDNN_ARG_WEIGHTS_LAYER   MKLDNN_ARG_WEIGHTS_0
#define MKLDNN_ARG_WORKSPACE   64