Searched refs:is_1x1 (Results 1 – 6 of 6) sorted by relevance
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_brgemm_conv_utils.cpp | 500 if (is_1x1 && is_amx(isa)) { in select_ic_block() 591 if (!is_1x1) M_tail = brgM_tail = (oh * ow) % sp_block; in estimate_brgemm_ur() 632 if (is_1x1 && is_amx(isa) && M > 0 && M_tail > 0) { in estimate_brgemm_ur() 660 if ((utils::one_of(exec_type, exec_trans, exec_vpad) || is_1x1) in get_brgemm_ur() 1167 if (is_1x1 && is_amx(isa)) return true; in fast_check_oc_block_1x1() 1616 jcp.is_1x1 = jcp.f_pad <= 0 && jcp.back_pad <= 0 && jcp.t_pad <= 0 in init_jcp() 1639 && IMPLICATION(is_amx(jcp.isa), !jcp.is_1x1); in init_jcp() 1702 jcp.is_ic_padded = !jcp.is_1x1 && one_of(jcp.wei_dt, bf16, s8) in init_jcp() 1719 if (jcp.is_1x1) return status::unimplemented; in init_conf() 1956 if (!jcp.is_1x1) return status::unimplemented; in init_1x1_conf()
|
H A D | jit_primitive_conf.hpp | 912 bool is_1x1; member
|
/dports/math/spectra/spectra-1.0.0/include/Spectra/LinAlg/ |
H A D | BKLDLT.h | 414 bool is_1x1 = permutate_mat(k, alpha); 417 if (is_1x1)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_brgemm_conv_utils.cpp | 331 bool is_1x1) const; 477 const memory_desc_t &dst_md, bool is_1x1) const { in get_brgemm_ur() 504 if ((utils::one_of(exec_type, exec_trans, exec_vpad) || is_1x1) in get_brgemm_ur() 1460 const bool is_1x1 = false; in init_conf() local 1461 const auto ur = cur_brgb.get_brgemm_ur(&attr, dst_md, is_1x1); in init_conf() 1645 const bool is_1x1 = true; in init_1x1_conf() local 1646 const auto ur = cur_brgb.get_brgemm_ur(&attr, dst_md, is_1x1); in init_1x1_conf()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/ |
H A D | xe_hp_convolution.cpp | 53 bool is_1x1 = (conf.kd * conf.kh * conf.kw == 1); in xe_hp_convolution_data_common_init_conf() local 58 if (is_1x1 && is_stride1 && is_eq_oi) { in xe_hp_convolution_data_common_init_conf()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
H A D | config.hpp | 113 bool is_1x1 = (kd * kh * kw == 1); in try_reduce_to_1d() local 116 if (is_1x1 && is_stride1 && is_eq_oi) { in try_reduce_to_1d()
|