| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | layer_normalization_pd.hpp | 223 if (arg == DNNL_ARG_DIFF_SCALE_SHIFT && use_scaleshift()) in arg_usage()
243 case DNNL_ARG_DIFF_SCALE_SHIFT: in arg_md()
|
| H A D | batch_normalization_pd.hpp | 229 if (arg == DNNL_ARG_DIFF_SCALE_SHIFT && use_scaleshift()) in arg_usage()
248 case DNNL_ARG_DIFF_SCALE_SHIFT: in arg_md()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | layer_normalization_pd.hpp | 223 if (arg == DNNL_ARG_DIFF_SCALE_SHIFT && use_scaleshift()) in arg_usage()
243 case DNNL_ARG_DIFF_SCALE_SHIFT: in arg_md()
|
| H A D | batch_normalization_pd.hpp | 229 if (arg == DNNL_ARG_DIFF_SCALE_SHIFT && use_scaleshift()) in arg_usage()
248 case DNNL_ARG_DIFF_SCALE_SHIFT: in arg_md()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | ref_layer_normalization.cpp | 218 conf.use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
220 = CTX_OUT_STORAGE(conf.use_scaleshift ? DNNL_ARG_DIFF_SCALE_SHIFT in execute_backward()
|
| H A D | gen9_batch_normalization.cpp | 406 conf.diff_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | ref_batch_normalization.cpp | 460 conf.diff_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | ref_layer_normalization.cpp | 324 conf.use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
326 = CTX_OUT_STORAGE(conf.use_scaleshift ? DNNL_ARG_DIFF_SCALE_SHIFT in execute_backward()
|
| H A D | gen9_batch_normalization.cpp | 411 conf.diff_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | ref_batch_normalization.cpp | 460 conf.diff_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | simple_layer_normalization.cpp | 182 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | ref_layer_normalization.cpp | 173 float *, DNNL_ARG_DIFF_SCALE_SHIFT, status) in execute_backward()
|
| H A D | ref_batch_normalization.cpp | 225 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | nspc_batch_normalization.cpp | 277 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
|
| H A D | ncsp_batch_normalization.cpp | 323 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | simple_layer_normalization.cpp | 183 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | ref_layer_normalization.cpp | 173 float *, DNNL_ARG_DIFF_SCALE_SHIFT, status) in execute_backward()
|
| H A D | ref_batch_normalization.cpp | 225 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT, in execute_backward()
|
| H A D | nspc_batch_normalization.cpp | 277 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
|
| H A D | ncsp_batch_normalization.cpp | 323 use_scale ? DNNL_ARG_DIFF_SCALE : DNNL_ARG_DIFF_SCALE_SHIFT); in execute_backward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_layer_normalization.cpp | 205 query::exec_arg_md, DNNL_ARG_DIFF_SCALE_SHIFT) in Backward()
266 args.insert({DNNL_ARG_DIFF_SCALE_SHIFT, diff_weights}); in execlnormBwd()
|
| H A D | test_batch_normalization_common.hpp | 290 query::exec_arg_md, DNNL_ARG_DIFF_SCALE_SHIFT) in Backward()
374 args.insert({DNNL_ARG_DIFF_SCALE_SHIFT, diff_weights}); in execBnormBwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_layer_normalization.cpp | 205 query::exec_arg_md, DNNL_ARG_DIFF_SCALE_SHIFT) in Backward()
266 args.insert({DNNL_ARG_DIFF_SCALE_SHIFT, diff_weights}); in execlnormBwd()
|
| H A D | test_batch_normalization_common.hpp | 290 query::exec_arg_md, DNNL_ARG_DIFF_SCALE_SHIFT) in Backward()
374 args.insert({DNNL_ARG_DIFF_SCALE_SHIFT, diff_weights}); in execBnormBwd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/ |
| H A D | mkldnn_dnnl_mangling.h | 34 #define MKLDNN_ARG_DIFF_SCALE_SHIFT DNNL_ARG_DIFF_SCALE_SHIFT
|