Home
last modified time | relevance | path

Searched refs:dat_tag_nCx8c (Results 1 – 12 of 12) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_sse41_conv_kernel_f32.cpp437 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 Djit_avx2_conv_kernel_f32.cpp645 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 Djit_avx2_1x1_conv_kernel_f32.cpp768 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 Djit_avx512_core_bf16_conv_kernel.cpp904 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 Djit_avx512_common_conv_kernel.cpp1409 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 Djit_sse41_conv_kernel_f32.cpp430 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 Djit_avx2_conv_kernel_f32.cpp635 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 Djit_avx2_1x1_conv_kernel_f32.cpp759 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 Djit_avx512_core_bf16_conv_kernel.cpp892 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 Djit_avx512_common_conv_kernel.cpp1396 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 Djit_sve_512_conv_kernel.cpp2087 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 Djit_sve_512_conv_kernel.cpp2087 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()