Home
last modified time | relevance | path

Searched refs:is_data_layout_nxc (Results 1 – 24 of 24) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx2_1x1_conv_kernel_f32.cpp771 const bool is_data_layout_nxc in init_conf() local
808 && IMPLICATION(!is_data_layout_nxc, in init_conf()
844 * (is_data_layout_nxc ? 1 : jcp.is) * sizeof(float); in init_conf()
859 load_blocking = is_data_layout_nxc in init_conf()
862 load_blocking_max = is_data_layout_nxc ? jcp.load_dim : 144; in init_conf()
865 reduce_blocking = is_data_layout_nxc ? jcp.reduce_dim in init_conf()
893 load_blocking = is_data_layout_nxc in init_conf()
900 reduce_blocking = is_data_layout_nxc ? jcp.reduce_dim in init_conf()
934 = IMPLICATION(is_data_layout_nxc, no_load_tail); in init_conf()
951 const int bcast_blocking_lim = is_data_layout_nxc ? 17 : 9; in init_conf()
[all …]
H A Djit_uni_dw_conv_kernel_utils.hpp136 const bool is_data_layout_nxc = data_tag == nxc_tag; in init_conf() local
190 if (is_data_layout_nxc) { in init_conf()
253 const bool ok_to_pad_channels = true && !is_data_layout_nxc in init_conf()
263 && IMPLICATION(!is_data_layout_nxc, jcp.ngroups % simd_w == 0) in init_conf()
387 bool is_data_layout_nxc in init_conf() local
389 auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
448 jcp.loop_order = is_data_layout_nxc ? loop_nhwcg : loop_ngcw; in init_conf()
580 bool is_data_layout_nxc in init_conf() local
583 auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
622 bool ok_to_pad_channels = !is_data_layout_nxc in init_conf()
[all …]
H A Djit_avx512_common_1x1_conv_kernel.cpp841 bool is_data_layout_nxc in init_conf() local
866 && IMPLICATION(!is_data_layout_nxc, in init_conf()
899 && !is_data_layout_nxc in init_conf()
920 if (!is_data_layout_nxc) { in init_conf()
1045 if (is_data_layout_nxc) { in init_conf()
1073 if (!is_data_layout_nxc in init_conf()
1302 * (is_data_layout_nxc ? jcp.ic : jcp.ic_block); in init_conf()
1304 * (is_data_layout_nxc ? jcp.oc : jcp.oc_block); in init_conf()
1311 * (is_data_layout_nxc ? 1 in init_conf()
1318 * (is_data_layout_nxc ? 1 : jcp.os); in init_conf()
[all …]
H A Djit_avx2_conv_kernel_f32.cpp660 bool is_data_layout_nxc in init_conf() local
665 if (is_data_layout_nxc && ndims == 3 && jcp.ow > 11 * 1024) in init_conf()
770 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
771 jcp.oc_tail = is_data_layout_nxc in init_conf()
1166 bool is_data_layout_nxc in init_conf() local
1182 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
1183 jcp.oc_tail = is_data_layout_nxc ? jcp.oc % simd_w : 0; in init_conf()
1199 && IMPLICATION(!is_data_layout_nxc, in init_conf()
1367 bool is_data_layout_nxc in init_conf() local
1404 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
[all …]
H A Djit_avx512_core_bf16_1x1_conv_kernel.cpp1292 bool is_data_layout_nxc in init_conf() local
1319 && IMPLICATION(!is_data_layout_nxc, in init_conf()
1394 * (is_data_layout_nxc ? 1 : jcp.bcast_dim); in init_conf()
1460 * (is_data_layout_nxc ? jcp.reduce_dim : jcp.reduce_block); in init_conf()
1474 if (is_data_layout_nxc in init_conf()
1610 jcp.uses_permw_transposition = !is_data_layout_nxc && jcp.oh > 14 in init_conf()
1642 * (is_data_layout_nxc && jcp.uses_permw_transposition in init_conf()
1646 * (is_data_layout_nxc && jcp.uses_permw_transposition in init_conf()
1661 ? (is_data_layout_nxc ? 1 : jcp.os) in init_conf()
1682 if (!is_data_layout_nxc) { in init_conf()
[all …]
H A Djit_avx512_core_bf16_conv_kernel.cpp910 bool is_data_layout_nxc in init_conf() local
920 && IMPLICATION(is_data_layout_nxc, in init_conf()
943 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
1013 if (is_data_layout_nxc) in init_conf()
1620 bool is_data_layout_nxc in init_conf() local
1628 && IMPLICATION(is_data_layout_nxc, in init_conf()
1651 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
4211 bool is_data_layout_nxc in init_conf() local
4224 auto src_tag = is_data_layout_nxc in init_conf()
4345 const bool nt_stores_ok = is_data_layout_nxc in init_conf()
[all …]
H A Djit_avx512_common_conv_kernel.cpp1415 bool is_data_layout_nxc in init_conf() local
1427 && IMPLICATION(is_data_layout_nxc, in init_conf()
1450 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
1493 src_tag = is_data_layout_nxc in init_conf()
1916 if (is_data_layout_nxc) { in init_conf()
2837 bool is_data_layout_nxc in init_conf() local
2849 && IMPLICATION(is_data_layout_nxc, in init_conf()
2874 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
2894 dat_tag = is_data_layout_nxc in init_conf()
3203 if (is_data_layout_nxc) { in init_conf()
[all …]
H A Djit_sse41_1x1_conv_kernel_f32.cpp626 const bool is_data_layout_nxc in init_conf() local
628 const auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
674 * (is_data_layout_nxc ? 1 : jcp.is) * sizeof(float); in init_conf()
679 * (is_data_layout_nxc ? jcp.oc : jcp.oc_block) * sizeof(float); in init_conf()
682 * (is_data_layout_nxc ? jcp.ic : jcp.ic_block) * sizeof(float); in init_conf()
H A Djit_sse41_conv_kernel_f32.cpp449 const bool is_data_layout_nxc in init_conf() local
501 = is_data_layout_nxc ? 1 : 4; /* the optimal value for the kernel */ in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx2_1x1_conv_kernel_f32.cpp762 const bool is_data_layout_nxc in init_conf() local
799 && IMPLICATION(!is_data_layout_nxc, in init_conf()
835 * (is_data_layout_nxc ? 1 : jcp.is) * sizeof(float); in init_conf()
850 load_blocking = is_data_layout_nxc in init_conf()
853 load_blocking_max = is_data_layout_nxc ? jcp.load_dim : 144; in init_conf()
856 reduce_blocking = is_data_layout_nxc ? jcp.reduce_dim in init_conf()
884 load_blocking = is_data_layout_nxc in init_conf()
891 reduce_blocking = is_data_layout_nxc ? jcp.reduce_dim in init_conf()
925 = IMPLICATION(is_data_layout_nxc, no_load_tail); in init_conf()
942 const int bcast_blocking_lim = is_data_layout_nxc ? 17 : 9; in init_conf()
[all …]
H A Djit_uni_dw_conv_kernel_utils.cpp89 const bool is_data_layout_nxc = data_tag == nxc_tag; in init_conf() local
143 if (is_data_layout_nxc) { in init_conf()
208 const bool ok_to_pad_channels = true && !is_data_layout_nxc in init_conf()
218 && IMPLICATION(!is_data_layout_nxc, jcp.ngroups % simd_w == 0) in init_conf()
307 bool is_data_layout_nxc in init_conf() local
309 auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
368 jcp.loop_order = is_data_layout_nxc ? loop_nhwcg : loop_ngcw; in init_conf()
461 bool is_data_layout_nxc in init_conf() local
464 auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
503 bool ok_to_pad_channels = !is_data_layout_nxc in init_conf()
[all …]
H A Djit_avx512_common_1x1_conv_kernel.cpp820 bool is_data_layout_nxc in init_conf() local
844 && IMPLICATION(!is_data_layout_nxc, in init_conf()
877 && !is_data_layout_nxc in init_conf()
898 if (!is_data_layout_nxc) { in init_conf()
1023 if (is_data_layout_nxc) { in init_conf()
1051 if (!is_data_layout_nxc in init_conf()
1280 * (is_data_layout_nxc ? jcp.ic : jcp.ic_block); in init_conf()
1282 * (is_data_layout_nxc ? jcp.oc : jcp.oc_block); in init_conf()
1289 * (is_data_layout_nxc ? 1 in init_conf()
1296 * (is_data_layout_nxc ? 1 : jcp.os); in init_conf()
[all …]
H A Djit_avx2_conv_kernel_f32.cpp650 bool is_data_layout_nxc in init_conf() local
655 if (is_data_layout_nxc && ndims == 3 && jcp.ow > 11 * 1024) in init_conf()
760 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
761 jcp.oc_tail = is_data_layout_nxc in init_conf()
1156 bool is_data_layout_nxc in init_conf() local
1172 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
1173 jcp.oc_tail = is_data_layout_nxc ? jcp.oc % simd_w : 0; in init_conf()
1189 && IMPLICATION(!is_data_layout_nxc, in init_conf()
1357 bool is_data_layout_nxc in init_conf() local
1394 jcp.ic_tail = is_data_layout_nxc ? jcp.ic % simd_w : 0; in init_conf()
[all …]
H A Djit_avx512_core_bf16_1x1_conv_kernel.cpp1273 bool is_data_layout_nxc in init_conf() local
1300 && IMPLICATION(!is_data_layout_nxc, in init_conf()
1375 * (is_data_layout_nxc ? 1 : jcp.bcast_dim); in init_conf()
1441 * (is_data_layout_nxc ? jcp.reduce_dim : jcp.reduce_block); in init_conf()
1455 if (is_data_layout_nxc in init_conf()
1591 jcp.uses_permw_transposition = !is_data_layout_nxc && jcp.oh > 14 in init_conf()
1623 * (is_data_layout_nxc && jcp.uses_permw_transposition in init_conf()
1627 * (is_data_layout_nxc && jcp.uses_permw_transposition in init_conf()
1642 ? (is_data_layout_nxc ? 1 : jcp.os) in init_conf()
1663 if (!is_data_layout_nxc) { in init_conf()
[all …]
H A Djit_avx512_core_bf16_conv_kernel.cpp898 bool is_data_layout_nxc in init_conf() local
908 && IMPLICATION(is_data_layout_nxc, in init_conf()
931 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
1001 if (is_data_layout_nxc) in init_conf()
1611 bool is_data_layout_nxc in init_conf() local
1619 && IMPLICATION(is_data_layout_nxc, in init_conf()
1642 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
4202 bool is_data_layout_nxc in init_conf() local
4215 auto src_tag = is_data_layout_nxc in init_conf()
4336 const bool nt_stores_ok = is_data_layout_nxc in init_conf()
[all …]
H A Djit_avx512_common_conv_kernel.cpp1402 bool is_data_layout_nxc in init_conf() local
1414 && IMPLICATION(is_data_layout_nxc, in init_conf()
1437 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
1458 src_tag = is_data_layout_nxc in init_conf()
1904 if (is_data_layout_nxc) { in init_conf()
2825 bool is_data_layout_nxc in init_conf() local
2837 && IMPLICATION(is_data_layout_nxc, in init_conf()
2862 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
2882 dat_tag = is_data_layout_nxc in init_conf()
3191 if (is_data_layout_nxc) { in init_conf()
[all …]
H A Djit_sse41_1x1_conv_kernel_f32.cpp619 const bool is_data_layout_nxc in init_conf() local
621 const auto dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_blocked; in init_conf()
667 * (is_data_layout_nxc ? 1 : jcp.is) * sizeof(float); in init_conf()
672 * (is_data_layout_nxc ? jcp.oc : jcp.oc_block) * sizeof(float); in init_conf()
675 * (is_data_layout_nxc ? jcp.ic : jcp.ic_block) * sizeof(float); in init_conf()
H A Djit_sse41_conv_kernel_f32.cpp442 const bool is_data_layout_nxc in init_conf() local
494 = is_data_layout_nxc ? 1 : 4; /* the optimal value for the kernel */ in init_conf()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_sve_512_conv_kernel.cpp892 bool is_data_layout_nxc in init_conf() local
1132 if (is_data_layout_nxc) { in init_conf()
2093 bool is_data_layout_nxc in init_conf() local
2113 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
2124 dat_tag = is_data_layout_nxc ? nxc_tag in init_conf()
2129 dat_tag = is_data_layout_nxc ? nxc_tag in init_conf()
2133 dat_tag = is_data_layout_nxc in init_conf()
2353 if (is_data_layout_nxc) { in init_conf()
3984 bool is_data_layout_nxc in init_conf() local
4086 && IMPLICATION(!is_data_layout_nxc, in init_conf()
[all …]
H A Djit_uni_dw_conv_kernel_utils.hpp137 const bool is_data_layout_nxc = data_tag == nxc_tag; in init_conf() local
186 if (is_data_layout_nxc) { in init_conf()
H A Djit_sve_512_1x1_conv_kernel.cpp720 bool is_data_layout_nxc in init_conf() local
722 auto required_dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx16c; in init_conf()
724 if (is_data_layout_nxc) return status::unimplemented; in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_sve_512_conv_kernel.cpp892 bool is_data_layout_nxc in init_conf() local
1132 if (is_data_layout_nxc) { in init_conf()
2093 bool is_data_layout_nxc in init_conf() local
2113 if (!IMPLICATION(!is_data_layout_nxc, in init_conf()
2124 dat_tag = is_data_layout_nxc ? nxc_tag in init_conf()
2129 dat_tag = is_data_layout_nxc ? nxc_tag in init_conf()
2133 dat_tag = is_data_layout_nxc in init_conf()
2353 if (is_data_layout_nxc) { in init_conf()
3984 bool is_data_layout_nxc in init_conf() local
4086 && IMPLICATION(!is_data_layout_nxc, in init_conf()
[all …]
H A Djit_uni_dw_conv_kernel_utils.hpp137 const bool is_data_layout_nxc = data_tag == nxc_tag; in init_conf() local
186 if (is_data_layout_nxc) { in init_conf()
H A Djit_sve_512_1x1_conv_kernel.cpp720 bool is_data_layout_nxc in init_conf() local
722 auto required_dat_tag = is_data_layout_nxc ? dat_tag_nxc : dat_tag_nCx16c; in init_conf()
724 if (is_data_layout_nxc) return status::unimplemented; in init_conf()