/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_sse41_1x1_convolution.cpp | 140 auto inner_ker = [&](int ocb, int icb, int n, int g, int oh, int ow, int ih, in execute_forward_thr() local 192 inner_ker(ocb, icb, n, g, oh, ow, ih, iw); in execute_forward_thr()
|
H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 550 auto inner_ker = [&](int icb, int ocb, int n, int g, int od, int oh, int ow, in execute_backward_data_thr() local 601 inner_ker(icb, ocb, n, g, od, oh, ow, id, ih, iw); in execute_backward_data_thr()
|
H A D | gemm_bf16_convolution.cpp | 582 auto inner_ker = [&](const int ic, const int oc, const int groups, in execute_forward_ncsp() local 668 inner_ker(ic, oc, g, od, nb_os, _src, _weights, _col, _dst_im, in execute_forward_ncsp()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_sse41_1x1_convolution.cpp | 140 auto inner_ker = [&](int ocb, int icb, int n, int g, int oh, int ow, int ih, in execute_forward_thr() local 192 inner_ker(ocb, icb, n, g, oh, ow, ih, iw); in execute_forward_thr()
|
H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 552 auto inner_ker = [&](int icb, int ocb, int n, int g, int od, int oh, int ow, in execute_backward_data_thr() local 603 inner_ker(icb, ocb, n, g, od, oh, ow, id, ih, iw); in execute_backward_data_thr()
|
H A D | gemm_bf16_convolution.cpp | 572 auto inner_ker = [&](const int ic, const int oc, const int groups, in execute_forward_ncsp() local 660 inner_ker(ic, oc, g, od, nb_os, _src, _weights, _col, _dst_im, in execute_forward_ncsp()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | gemm_convolution.cpp | 254 auto inner_ker = [&](int spatial, const im_pos_t &curr, im_pos_t &prev, in execute_forward_ncsp() local 388 = inner_ker(spatial, curr, prev, step, end); in execute_forward_ncsp() 403 = inner_ker(spatial, curr, prev, step, end); in execute_forward_ncsp()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | gemm_convolution.cpp | 251 auto inner_ker = [&](int spatial, const im_pos_t &curr, im_pos_t &prev, in execute_forward_ncsp() local 385 = inner_ker(spatial, curr, prev, step, end); in execute_forward_ncsp() 400 = inner_ker(spatial, curr, prev, step, end); in execute_forward_ncsp()
|