mxnet
|
#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 |