/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_rnn_forward.cpp | 118 ASSERT_TRUE(pd.query_md(query::exec_arg_md, DNNL_ARG_BIAS) in testExecArgQueries() 371 {DNNL_ARG_BIAS, bias_ref}, in Test() 395 {DNNL_ARG_BIAS, bias_tgt}, in Test()
|
H A D | test_inner_product_forward.cpp | 204 ip_primitive_desc.query_md(query::exec_arg_md, DNNL_ARG_BIAS) in Test() 210 {DNNL_ARG_BIAS, ip_bias}, in Test()
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_rnn_forward.cpp | 118 ASSERT_TRUE(pd.query_md(query::exec_arg_md, DNNL_ARG_BIAS) in testExecArgQueries() 371 {DNNL_ARG_BIAS, bias_ref}, in Test() 395 {DNNL_ARG_BIAS, bias_tgt}, in Test()
|
H A D | test_iface_attr.cpp | 155 const std::vector<int> unsupported_args = {DNNL_ARG_BIAS, DNNL_ARG_DST_2, in TEST_F() 210 const std::vector<int> unsupported_args = {DNNL_ARG_BIAS, DNNL_ARG_DST_2, in HANDLE_EXCEPTIONS_FOR_TEST_F()
|
H A D | test_inner_product_forward.cpp | 204 ip_primitive_desc.query_md(query::exec_arg_md, DNNL_ARG_BIAS) in Test() 210 {DNNL_ARG_BIAS, ip_bias}, in Test()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | ref_inner_product.cpp | 34 auto bias = CTX_IN_MEM(const void *, DNNL_ARG_BIAS); in execute_forward()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx2_1x1_convolution.cpp | 45 auto bias = CTX_IN_MEM(const data_t *, DNNL_ARG_BIAS); in execute_forward() 50 const data_t *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward() 94 pd()->arg_md(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS)); in execute_forward_thr()
|
H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 43 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward() 48 const char *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 48 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward() 53 const char *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_avx512_common_convolution_winograd.hpp | 151 auto bias = CTX_IN_MEM(const float *, DNNL_ARG_BIAS); in execute()
|
H A D | gemm_bf16_inner_product.cpp | 47 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_avx512_core_amx_1x1_convolution.cpp | 70 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_avx2_1x1_convolution.cpp | 45 auto bias = CTX_IN_MEM(const data_t *, DNNL_ARG_BIAS); in execute_forward() 50 const data_t *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward() 94 pd()->arg_md(DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS)); in execute_forward_thr()
|
H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 47 const auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward() 52 const char *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 42 const auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward() 47 const char *, DNNL_ARG_ATTR_POST_OP_DW | DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_avx512_core_amx_1x1_convolution.cpp | 67 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward()
|
H A D | jit_avx512_common_convolution_winograd.hpp | 150 auto bias = CTX_IN_MEM(const float *, DNNL_ARG_BIAS); in execute()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_uni_dw_convolution.cpp | 51 auto bias_in = CTX_IN_MEM(const bf16_data_t *, DNNL_ARG_BIAS); in execute_forward() 58 auto bias_in = CTX_IN_MEM(const f32_data_t *, DNNL_ARG_BIAS); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_uni_dw_convolution.cpp | 51 auto bias_in = CTX_IN_MEM(const bf16_data_t *, DNNL_ARG_BIAS); in execute_forward() 58 auto bias_in = CTX_IN_MEM(const f32_data_t *, DNNL_ARG_BIAS); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
H A D | xe_hp_convolution.cpp | 246 auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_forward() 432 auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_backward_data()
|
H A D | xe_hpc_1x1_convolution.cpp | 204 auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/conv/ |
H A D | ref_conv.cpp | 26 const dnn_mem_t &bia_m = args.find(DNNL_ARG_BIAS); in compute_ref_direct_fwd() 102 const dnn_mem_t &bia_m = args.find(DNNL_ARG_BIAS); in compute_ref_direct_bwd_d()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
H A D | cudnn_inner_product.cpp | 48 CTX_IN_ACCESSOR(DNNL_ARG_BIAS)); in execute()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
H A D | cudnn_inner_product.cpp | 46 CTX_IN_ACCESSOR(DNNL_ARG_BIAS)); in execute()
|
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/ |
H A D | OneDNNConvolution.cpp | 159 {DNNL_ARG_BIAS, conv_bias}, in onExecute()
|