Home
last modified time | relevance | path

Searched refs:curr_dst_tag (Results 1 – 8 of 8) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_dw_conv_kernel_utils.cpp305 auto curr_dst_tag in init_conf() local
308 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
322 } else if (curr_dst_tag != dat_tag) in init_conf()
458 auto curr_dst_tag in init_conf() local
462 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
477 } else if (curr_dst_tag != dat_tag) in init_conf()
H A Djit_avx512_core_bf16_conv_kernel.cpp896 auto curr_dst_tag = dst_d.matches_one_of_tag( in init_conf() local
899 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
964 else if (curr_dst_tag != dst_tag) in init_conf()
1609 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
1612 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
1670 } else if (curr_dst_tag != dat_tag) in init_conf()
4200 auto curr_dst_tag in init_conf() local
4203 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
4233 } else if (curr_dst_tag != dst_tag) in init_conf()
H A Djit_avx512_common_conv_kernel.cpp1400 auto curr_dst_tag = dst_d.matches_one_of_tag( in init_conf() local
1403 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
1473 else if (curr_dst_tag != dst_tag) in init_conf()
2823 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
2826 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
2897 } else if (curr_dst_tag != dat_tag) in init_conf()
5753 auto curr_dst_tag in init_conf() local
5756 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
5804 } else if (curr_dst_tag != dst_tag) in init_conf()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_dw_conv_kernel_utils.hpp385 auto curr_dst_tag in init_conf() local
388 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
402 } else if (curr_dst_tag != dat_tag) in init_conf()
577 auto curr_dst_tag in init_conf() local
581 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
596 } else if (curr_dst_tag != dat_tag) in init_conf()
H A Djit_avx512_core_bf16_conv_kernel.cpp908 auto curr_dst_tag = dst_d.matches_one_of_tag( in init_conf() local
911 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
976 else if (curr_dst_tag != dst_tag) in init_conf()
1618 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
1621 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
1679 } else if (curr_dst_tag != dat_tag) in init_conf()
4209 auto curr_dst_tag in init_conf() local
4212 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
4242 } else if (curr_dst_tag != dst_tag) in init_conf()
H A Djit_avx512_common_conv_kernel.cpp1413 auto curr_dst_tag = dst_d.matches_one_of_tag( in init_conf() local
1416 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
1508 else if (curr_dst_tag != dst_tag) in init_conf()
2835 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
2838 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
2909 } else if (curr_dst_tag != dat_tag) in init_conf()
5752 auto curr_dst_tag in init_conf() local
5755 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
5803 } else if (curr_dst_tag != dst_tag) in init_conf()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_sve_512_conv_kernel.cpp891 auto curr_dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx16c); in init_conf() local
893 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
950 else if (curr_dst_tag != dst_tag) in init_conf()
2091 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
2094 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
2148 } else if (curr_dst_tag != dat_tag) in init_conf()
3982 auto curr_dst_tag in init_conf() local
3985 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
4035 } else if (curr_dst_tag != dst_tag) in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_sve_512_conv_kernel.cpp891 auto curr_dst_tag = dst_d.matches_one_of_tag(dat_tag_nxc, dat_tag_nCx16c); in init_conf() local
893 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
950 else if (curr_dst_tag != dst_tag) in init_conf()
2091 auto curr_dst_tag = diff_dst_d.matches_one_of_tag( in init_conf() local
2094 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
2148 } else if (curr_dst_tag != dat_tag) in init_conf()
3982 auto curr_dst_tag in init_conf() local
3985 = utils::everyone_is(dat_tag_nxc, curr_src_tag, curr_dst_tag); in init_conf()
4035 } else if (curr_dst_tag != dst_tag) in init_conf()