/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx2_1x1_conv_kernel_f32.cpp | 771 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 D | jit_uni_dw_conv_kernel_utils.hpp | 136 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 D | jit_avx512_common_1x1_conv_kernel.cpp | 841 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 D | jit_avx2_conv_kernel_f32.cpp | 660 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 D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 1292 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 D | jit_avx512_core_bf16_conv_kernel.cpp | 910 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 D | jit_avx512_common_conv_kernel.cpp | 1415 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 D | jit_sse41_1x1_conv_kernel_f32.cpp | 626 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 D | jit_sse41_conv_kernel_f32.cpp | 449 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 D | jit_avx2_1x1_conv_kernel_f32.cpp | 762 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 D | jit_uni_dw_conv_kernel_utils.cpp | 89 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 D | jit_avx512_common_1x1_conv_kernel.cpp | 820 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 D | jit_avx2_conv_kernel_f32.cpp | 650 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 D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 1273 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 D | jit_avx512_core_bf16_conv_kernel.cpp | 898 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 D | jit_avx512_common_conv_kernel.cpp | 1402 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 D | jit_sse41_1x1_conv_kernel_f32.cpp | 619 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 D | jit_sse41_conv_kernel_f32.cpp | 442 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 D | jit_sve_512_conv_kernel.cpp | 892 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 D | jit_uni_dw_conv_kernel_utils.hpp | 137 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 D | jit_sve_512_1x1_conv_kernel.cpp | 720 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 D | jit_sve_512_conv_kernel.cpp | 892 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 D | jit_uni_dw_conv_kernel_utils.hpp | 137 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 D | jit_sve_512_1x1_conv_kernel.cpp | 720 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()
|