/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_sse41_conv_kernel_f32.cpp | 437 const auto dat_tag_nCx8c = (ndims == 3 ? nCw8c : nChw8c); in init_conf() local 445 = src_d.matches_one_of_tag(dat_tag_ncx, dat_tag_nxc, dat_tag_nCx8c); in init_conf() 447 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 480 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 485 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() 486 && jcp.dst_tag == dat_tag_nCx8c) in init_conf()
|
H A D | jit_avx2_conv_kernel_f32.cpp | 645 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 655 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 713 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 718 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() 719 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 1154 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1159 jcp.src_tag = diff_src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 1354 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1411 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 1416 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() [all …]
|
H A D | jit_avx2_1x1_conv_kernel_f32.cpp | 768 const auto dat_tag_nCx8c = utils::pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 769 jcp.src_tag = src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 770 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 773 const auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 904 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 907 dat_tag_nCx8c, dat_tag_nCx4c, dat_tag_ncx); in init_conf() 909 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 951 dst_tag = src_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf() 1614 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1617 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1619 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1660 dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf()
|
H A D | jit_avx512_common_conv_kernel.cpp | 1409 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1412 dat_tag_nCx8c, dat_tag_nCx4c, dat_tag_ncx); in init_conf() 1414 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1483 src_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf() 2831 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 2834 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 2836 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_sse41_conv_kernel_f32.cpp | 430 const auto dat_tag_nCx8c = (ndims == 3 ? nCw8c : nChw8c); in init_conf() local 438 = src_d.matches_one_of_tag(dat_tag_ncx, dat_tag_nxc, dat_tag_nCx8c); in init_conf() 440 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 473 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 478 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() 479 && jcp.dst_tag == dat_tag_nCx8c) in init_conf()
|
H A D | jit_avx2_conv_kernel_f32.cpp | 635 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 645 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 703 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 708 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() 709 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 1144 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1149 jcp.src_tag = diff_src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 1344 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1401 && jcp.dst_tag == dat_tag_nCx8c) in init_conf() 1406 && ((jcp.src_tag == dat_tag_nCx8c in init_conf() [all …]
|
H A D | jit_avx2_1x1_conv_kernel_f32.cpp | 759 const auto dat_tag_nCx8c = utils::pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 760 jcp.src_tag = src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 761 jcp.dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx8c); in init_conf() 764 const auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 892 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 895 dat_tag_nCx8c, dat_tag_nCx4c, dat_tag_ncx); in init_conf() 897 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 939 dst_tag = src_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf() 1605 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1608 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1610 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1651 dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf()
|
H A D | jit_avx512_common_conv_kernel.cpp | 1396 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 1399 dat_tag_nCx8c, dat_tag_nCx4c, dat_tag_ncx); in init_conf() 1401 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 1448 src_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx8c; in init_conf() 2819 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 2822 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 2824 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_sve_512_conv_kernel.cpp | 2087 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 2090 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 2092 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_sve_512_conv_kernel.cpp | 2087 const auto dat_tag_nCx8c = pick(ndims - 3, nCw8c, nChw8c, nCdhw8c); in init_conf() local 2090 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf() 2092 dat_tag_nxc, dat_tag_nCx16c, dat_tag_nCx8c, dat_tag_nCx4c); in init_conf()
|