/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_iface_runtime_attr.cpp | 80 for (auto arg : {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, in TEST_F() 144 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 177 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 220 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 243 DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 288 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in CPU_TEST_F() 290 || arg == DNNL_ARG_BIAS) { in CPU_TEST_F() 404 DNNL_ARG_WEIGHTS_LAYER, DNNL_ARG_WEIGHTS_ITER, DNNL_ARG_BIAS, in TEST_F()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | ref_fused_convolution.hpp | 113 case DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS: in arg_md() 123 if (arg == (DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS) in arg_usage() 202 arg_cache.append_ctx_arg(DNNL_ARG_BIAS); in init_ops() 263 arg_cache.append_ctx_arg(DNNL_ARG_BIAS, in init_ops() 264 DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in init_ops()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | ref_fused_convolution.hpp | 113 case DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS: in arg_md() 123 if (arg == (DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS) in arg_usage() 202 arg_cache.append_ctx_arg(DNNL_ARG_BIAS); in init_ops() 263 arg_cache.append_ctx_arg(DNNL_ARG_BIAS, in init_ops() 264 DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in init_ops()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
H A D | gemm_matmul.cpp | 33 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS); in execute() 49 gemm_args.bias = &CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute()
|
H A D | ref_matmul.cpp | 29 const auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_ref() 52 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS, pd()->weights_md(1)); in execute_ref()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
H A D | gemm_matmul.cpp | 33 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS); in execute() 49 gemm_args.bias = &CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute()
|
H A D | ref_matmul.cpp | 29 const auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_ref() 52 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS, pd()->weights_md(1)); in execute_ref()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_iface_runtime_attr.cpp | 80 for (auto arg : {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, in TEST_F() 144 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 177 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 225 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 260 DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in TEST_F() 305 {DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS, DNNL_ARG_DST}) { in CPU_TEST_F() 307 || arg == DNNL_ARG_BIAS) { in CPU_TEST_F() 417 DNNL_ARG_WEIGHTS_LAYER, DNNL_ARG_WEIGHTS_ITER, DNNL_ARG_BIAS, in TEST_F()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/conv/ |
H A D | conv_dw_fusion.cpp | 296 = prb->dir & FLAG_WEI ? q(DNNL_ARG_DIFF_BIAS) : q(DNNL_ARG_BIAS); in doit() 304 : q(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in doit() 360 = p0->dir & FLAG_WEI ? q0(DNNL_ARG_DIFF_BIAS) : q0(DNNL_ARG_BIAS); in doit() 415 = prb->dir & FLAG_WEI ? q1(DNNL_ARG_DIFF_BIAS) : q1(DNNL_ARG_BIAS); in doit() 462 args0.set(DNNL_ARG_BIAS, bia_dt0); in doit() 472 args1.set(DNNL_ARG_BIAS, bia_dt1); in doit() 508 args.set(DNNL_ARG_BIAS, bia_dt); in doit() 511 args.set(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS, fused_bia_dt); in doit()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/conv/ |
H A D | conv_dw_fusion.cpp | 297 = prb->dir & FLAG_WEI ? q(DNNL_ARG_DIFF_BIAS) : q(DNNL_ARG_BIAS); in doit() 305 : q(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in doit() 361 = p0->dir & FLAG_WEI ? q0(DNNL_ARG_DIFF_BIAS) : q0(DNNL_ARG_BIAS); in doit() 416 = prb->dir & FLAG_WEI ? q1(DNNL_ARG_DIFF_BIAS) : q1(DNNL_ARG_BIAS); in doit() 463 args0.set(DNNL_ARG_BIAS, bia_dt0); in doit() 473 args1.set(DNNL_ARG_BIAS, bia_dt1); in doit() 509 args.set(DNNL_ARG_BIAS, bia_dt); in doit() 512 args.set(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS, fused_bia_dt); in doit()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
H A D | cudnn_matmul_executor.hpp | 112 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 164 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 203 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 268 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
H A D | cudnn_matmul_executor.hpp | 114 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 166 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 205 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute() 268 auto bias_acc = CTX_IN_ACCESSOR(DNNL_ARG_BIAS); in execute()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | matmul_pd.hpp | 54 arg, DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS); in arg_usage() 66 case DNNL_ARG_BIAS: return weights_md(1); in arg_md()
|
H A D | rnn_pd.hpp | 216 if (arg == DNNL_ARG_BIAS && with_bias()) return arg_usage_t::input; in arg_usage() 244 case DNNL_ARG_BIAS: in arg_md() 297 if (arg == DNNL_ARG_BIAS) return arg_usage_t::input; in arg_usage() 344 case DNNL_ARG_BIAS: in arg_md()
|
H A D | deconvolution_pd.hpp | 176 if (arg == DNNL_ARG_BIAS && with_bias()) return arg_usage_t::input; in arg_usage() 187 case DNNL_ARG_BIAS: return weights_md(1); in arg_md() 243 case DNNL_ARG_BIAS: return weights_md(1); in arg_md()
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | matmul_pd.hpp | 54 arg, DNNL_ARG_SRC, DNNL_ARG_WEIGHTS, DNNL_ARG_BIAS); in arg_usage() 66 case DNNL_ARG_BIAS: return weights_md(1); in arg_md()
|
H A D | rnn_pd.hpp | 216 if (arg == DNNL_ARG_BIAS && with_bias()) return arg_usage_t::input; in arg_usage() 244 case DNNL_ARG_BIAS: in arg_md() 297 if (arg == DNNL_ARG_BIAS) return arg_usage_t::input; in arg_usage() 344 case DNNL_ARG_BIAS: in arg_md()
|
H A D | deconvolution_pd.hpp | 176 if (arg == DNNL_ARG_BIAS && with_bias()) return arg_usage_t::input; in arg_usage() 187 case DNNL_ARG_BIAS: return weights_md(1); in arg_md() 243 case DNNL_ARG_BIAS: return weights_md(1); in arg_md()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
H A D | cnn_inference_f32.cpp | 178 {DNNL_ARG_BIAS, conv1_user_bias_memory}, in simple_net() 309 {DNNL_ARG_BIAS, conv2_user_bias_memory}, in simple_net() 426 {DNNL_ARG_BIAS, conv3_user_bias_memory}, in simple_net() 500 {DNNL_ARG_BIAS, conv4_user_bias_memory}, in simple_net() 573 {DNNL_ARG_BIAS, conv5_user_bias_memory}, in simple_net() 665 {DNNL_ARG_BIAS, fc6_user_bias_memory}, in simple_net() 707 {DNNL_ARG_BIAS, fc7_user_bias_memory}, in simple_net() 750 {DNNL_ARG_BIAS, fc8_user_bias_memory}, in simple_net()
|
H A D | rnn_training_f32.cpp | 388 {DNNL_ARG_BIAS, common_bias_memory}, in simple_net() 401 {DNNL_ARG_BIAS, common_bias_memory}, in simple_net() 654 {DNNL_ARG_BIAS, common_bias_bwd_memory}, in simple_net() 673 {DNNL_ARG_BIAS, common_bias_bwd_memory}, in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/examples/ |
H A D | cnn_inference_f32.cpp | 178 {DNNL_ARG_BIAS, conv1_user_bias_memory}, in simple_net() 309 {DNNL_ARG_BIAS, conv2_user_bias_memory}, in simple_net() 426 {DNNL_ARG_BIAS, conv3_user_bias_memory}, in simple_net() 500 {DNNL_ARG_BIAS, conv4_user_bias_memory}, in simple_net() 573 {DNNL_ARG_BIAS, conv5_user_bias_memory}, in simple_net() 665 {DNNL_ARG_BIAS, fc6_user_bias_memory}, in simple_net() 707 {DNNL_ARG_BIAS, fc7_user_bias_memory}, in simple_net() 750 {DNNL_ARG_BIAS, fc8_user_bias_memory}, in simple_net()
|
H A D | rnn_training_f32.cpp | 388 {DNNL_ARG_BIAS, common_bias_memory}, in simple_net() 401 {DNNL_ARG_BIAS, common_bias_memory}, in simple_net() 654 {DNNL_ARG_BIAS, common_bias_bwd_memory}, in simple_net() 673 {DNNL_ARG_BIAS, common_bias_bwd_memory}, in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/matmul/ |
H A D | ref_matmul.cpp | 41 const auto bias = CTX_IN_MEM(const void *, DNNL_ARG_BIAS); in execute_ref() 50 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS, pd()->weights_md(1)); in execute_ref()
|
H A D | ref_matmul_int8.cpp | 42 const auto bias = CTX_IN_MEM(const void *, DNNL_ARG_BIAS); in execute_ref() 54 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS, pd()->weights_md(1)); in execute_ref()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/matmul/ |
H A D | ref_matmul.cpp | 43 const auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_ref() 54 const auto bia_d = ctx.memory_mdw(DNNL_ARG_BIAS, pd()->weights_md(1)); in execute_ref()
|