Home
last modified time | relevance | path

Searched refs:DNNL_ARG_BIAS (Results 1 – 25 of 258) sorted by relevance

1234567891011

/dports/math/onednn/oneDNN-2.5.1/tests/gtests/
H A Dtest_iface_runtime_attr.cpp80 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 Dref_fused_convolution.hpp113 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 Dref_fused_convolution.hpp113 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 Dgemm_matmul.cpp33 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 Dref_matmul.cpp29 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 Dgemm_matmul.cpp33 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 Dref_matmul.cpp29 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 Dtest_iface_runtime_attr.cpp80 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 Dconv_dw_fusion.cpp296 = 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 Dconv_dw_fusion.cpp297 = 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 Dcudnn_matmul_executor.hpp112 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 Dcudnn_matmul_executor.hpp114 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 Dmatmul_pd.hpp54 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 Drnn_pd.hpp216 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 Ddeconvolution_pd.hpp176 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 Dmatmul_pd.hpp54 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 Drnn_pd.hpp216 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 Ddeconvolution_pd.hpp176 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 Dcnn_inference_f32.cpp178 {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 Drnn_training_f32.cpp388 {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 Dcnn_inference_f32.cpp178 {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 Drnn_training_f32.cpp388 {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 Dref_matmul.cpp41 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 Dref_matmul_int8.cpp42 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 Dref_matmul.cpp43 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()

1234567891011