| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_sse41_1x1_convolution.cpp | 281 auto bcast_iter = bcast_start; in execute_forward_thr() local
282 while (bcast_iter < bcast_end) { in execute_forward_thr()
284 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
304 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 448 auto bcast_iter = bcast_start; in execute_forward_thr() local
449 while (bcast_iter < bcast_end) { in execute_forward_thr()
451 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
472 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 447 auto bcast_iter = bcast_start; in execute_forward_thr() local
448 while (bcast_iter < bcast_end) { in execute_forward_thr()
450 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
471 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx2_1x1_convolution.cpp | 325 auto bcast_iter = bcast_start; in execute_forward_thr() local
326 while (bcast_iter < bcast_end) { in execute_forward_thr()
328 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
349 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_common_1x1_convolution.cpp | 396 auto bcast_iter = bcast_start; in execute_forward_thr() local
397 while (bcast_iter < bcast_end) { in execute_forward_thr()
399 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
419 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 408 auto bcast_iter = bcast_start; in execute_forward_thr() local
409 while (bcast_iter < bcast_end) { in execute_forward_thr()
411 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
432 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_sse41_1x1_convolution.cpp | 281 auto bcast_iter = bcast_start; in execute_forward_thr() local
282 while (bcast_iter < bcast_end) { in execute_forward_thr()
284 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
304 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 451 auto bcast_iter = bcast_start; in execute_forward_thr() local
452 while (bcast_iter < bcast_end) { in execute_forward_thr()
454 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
475 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 451 auto bcast_iter = bcast_start; in execute_forward_thr() local
452 while (bcast_iter < bcast_end) { in execute_forward_thr()
454 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
475 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx2_1x1_convolution.cpp | 325 auto bcast_iter = bcast_start; in execute_forward_thr() local
326 while (bcast_iter < bcast_end) { in execute_forward_thr()
328 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
349 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_common_1x1_convolution.cpp | 396 auto bcast_iter = bcast_start; in execute_forward_thr() local
397 while (bcast_iter < bcast_end) { in execute_forward_thr()
399 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
419 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|
| H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 408 auto bcast_iter = bcast_start; in execute_forward_thr() local
409 while (bcast_iter < bcast_end) { in execute_forward_thr()
411 nd_iterator_init(bcast_iter, n, jcp.mb, g, jcp.ngroups, oh_dw, in execute_forward_thr()
432 bcast_iter += nb_bcast_blocking; in execute_forward_thr()
|