Home
last modified time | relevance | path

Searched refs:DNNL_ARG_BIAS (Results 126 – 150 of 258) sorted by relevance

1234567891011

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/
H A Dtest_rnn_forward.cpp118 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 Dtest_inner_product_forward.cpp204 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 Dtest_rnn_forward.cpp118 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 Dtest_iface_attr.cpp155 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 Dtest_inner_product_forward.cpp204 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 Dref_inner_product.cpp34 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 Djit_avx2_1x1_convolution.cpp45 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 Djit_avx512_core_x8s8s32x_1x1_convolution.cpp43 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 Djit_uni_x8s8s32x_1x1_convolution.cpp48 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 Djit_avx512_common_convolution_winograd.hpp151 auto bias = CTX_IN_MEM(const float *, DNNL_ARG_BIAS); in execute()
H A Dgemm_bf16_inner_product.cpp47 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward()
H A Djit_avx512_core_amx_1x1_convolution.cpp70 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 Djit_avx2_1x1_convolution.cpp45 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 Djit_uni_x8s8s32x_1x1_convolution.cpp47 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 Djit_avx512_core_x8s8s32x_1x1_convolution.cpp42 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 Djit_avx512_core_amx_1x1_convolution.cpp67 auto bias = CTX_IN_MEM(const char *, DNNL_ARG_BIAS); in execute_forward()
H A Djit_avx512_common_convolution_winograd.hpp150 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 Djit_uni_dw_convolution.cpp51 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 Djit_uni_dw_convolution.cpp51 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 Dxe_hp_convolution.cpp246 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 Dxe_hpc_1x1_convolution.cpp204 auto &bias = CTX_IN_STORAGE(DNNL_ARG_BIAS); in execute_forward()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/conv/
H A Dref_conv.cpp26 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 Dcudnn_inner_product.cpp48 CTX_IN_ACCESSOR(DNNL_ARG_BIAS)); in execute()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/
H A Dcudnn_inner_product.cpp46 CTX_IN_ACCESSOR(DNNL_ARG_BIAS)); in execute()
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/
H A DOneDNNConvolution.cpp159 {DNNL_ARG_BIAS, conv_bias}, in onExecute()

1234567891011