Home
last modified time | relevance | path

Searched refs:is_fast_postops (Results 1 – 6 of 6) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_core_amx_1x1_conv_kernel.hpp148 static bool is_fast_postops(const jit_conv_conf_t &jcp);
H A Djit_avx512_core_amx_1x1_conv_kernel.cpp244 bool jit_avx512_core_amx_1x1_fwd_kernel_t::is_fast_postops( in is_fast_postops() function in dnnl::impl::cpu::x64::jit_avx512_core_amx_1x1_fwd_kernel_t
584 if (!jcp.is_fast_postops) store_output_vector(zmm_r, ocb, oh_, ow_); in store_output()
586 if (do_store && jcp.is_fast_postops) store_output_vectors(ocb, osb); in store_output()
1030 jcp.is_fast_postops = is_fast_postops(jcp); in init_conf()
H A Djit_primitive_conf.hpp121 bool is_fast_postops; // maybe skip injector for sum and/or relu member
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx512_core_amx_1x1_conv_kernel.hpp151 static bool is_fast_postops(const jit_conv_conf_t &jcp);
H A Djit_avx512_core_amx_1x1_conv_kernel.cpp246 bool jit_avx512_core_amx_1x1_fwd_kernel_t::is_fast_postops( in is_fast_postops() function in dnnl::impl::cpu::x64::jit_avx512_core_amx_1x1_fwd_kernel_t
603 if (!jcp.is_fast_postops) store_output_vector(zmm_r, ocb, oh_, ow_); in store_output()
605 if (do_store && jcp.is_fast_postops) store_output_vectors(ocb, osb); in store_output()
1104 jcp.is_fast_postops = is_fast_postops(jcp); in init_conf()
H A Djit_primitive_conf.hpp123 bool is_fast_postops; // maybe skip injector for sum and/or relu member