Home
last modified time | relevance | path

Searched refs:apply_postops (Results 1 – 25 of 90) sorted by relevance

1234

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dgemm_inner_product_utils.cpp81 const bool apply_postops in operator ()() local
91 if (apply_postops) { in operator ()()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dgemm_inner_product_utils.cpp75 const bool apply_postops in operator ()() local
88 if (apply_postops) { in operator ()()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_resampling_kernel.cpp278 void jit_uni_resampling_kernel_t<isa, Vmm>::apply_postops( in apply_postops() function in dnnl::impl::cpu::x64::jit_uni_resampling_kernel_t
371 if (conf_.with_postops) apply_postops(vmm_src_.getIdx(), is_tail); in nearest_ncsp_format()
446 apply_postops(vmm_src_.getIdx(), is_tail, &reg_c); in nearest_c_oriented_format()
525 if (conf_.with_postops) apply_postops(vmm_idx(0), is_tail); in linear_ncsp_format()
619 apply_postops(src_ftl_.getIdx(), is_tail, &reg_c); in linear_c_oriented_format()
H A Djit_sse41_1x1_conv_kernel_f32.hpp88 void apply_postops(const int load_loop_blk, const int ur);
H A Djit_avx2_1x1_conv_kernel_f32.hpp90 void apply_postops(
H A Djit_avx512_common_1x1_conv_kernel.hpp93 void apply_postops(const bool is_out_layout_nxc, const int load_loop_blk,
H A Djit_sse41_conv_kernel_f32.hpp116 void apply_postops(const int oc_blocks, const int ur_w);
H A Djit_uni_binary_kernel.hpp142 void apply_postops(int unroll, bool tail);
H A Djit_uni_resampling_kernel.hpp88 void apply_postops(const int data_idx, const bool is_tail,
H A Djit_avx512_core_amx_1x1_conv_kernel.hpp145 void apply_postops(const Xbyak::Zmm &zmm_out, const float *p_sum_scale,
H A Djit_uni_x8s8s32x_1x1_conv_kernel.hpp113 void apply_postops(const int ur, const int load_loop_blk,
H A Djit_avx512_core_x8s8s32x_1x1_conv_kernel.hpp115 void apply_postops(const int load_loop_blk, const int ur,
H A Djit_gemm_x8s8s32x_convolution_utils.cpp49 void apply_postops(const Xbyak::Reg64 &reg_dst, const int idx);
327 void jit_pp_ker_t::apply_postops(const Xbyak::Reg64 &reg_dst, const int idx) { in apply_postops() function in dnnl::impl::cpu::x64::gemm_x8s8s32x_convolution_utils::jit_pp_ker_t
487 apply_postops(reg_dst_, idx); in generate()
H A Djit_avx512_core_bf16_1x1_conv_kernel.hpp129 void apply_postops(const int load_loop_blk, const int ur);
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_resampling_kernel.cpp278 void jit_uni_resampling_kernel_t<isa, Vmm>::apply_postops( in apply_postops() function in dnnl::impl::cpu::x64::jit_uni_resampling_kernel_t
356 if (conf_.with_postops) apply_postops(vmm_src_.getIdx(), is_tail); in nearest_ncsp_format()
431 apply_postops(vmm_src_.getIdx(), is_tail, &reg_c); in nearest_c_oriented_format()
510 if (conf_.with_postops) apply_postops(vmm_idx(0), is_tail); in linear_ncsp_format()
604 apply_postops(src_ftl_.getIdx(), is_tail, &reg_c); in linear_c_oriented_format()
H A Djit_sse41_1x1_conv_kernel_f32.hpp88 void apply_postops(const int load_loop_blk, const int ur);
H A Djit_avx2_1x1_conv_kernel_f32.hpp90 void apply_postops(
H A Djit_avx512_common_1x1_conv_kernel.hpp103 void apply_postops(const bool is_out_layout_nxc, const int load_loop_blk,
H A Djit_sse41_conv_kernel_f32.hpp117 void apply_postops(const int oc_blocks, const int ur_w);
H A Djit_uni_binary_kernel.hpp142 void apply_postops(int unroll, bool tail);
H A Djit_avx512_core_amx_1x1_conv_kernel.hpp148 void apply_postops(const Xbyak::Zmm &zmm_out, const float *p_sum_scale,
H A Djit_uni_x8s8s32x_1x1_conv_kernel.hpp116 void apply_postops(const int ur, const int load_loop_blk,
H A Djit_uni_resampling_kernel.hpp88 void apply_postops(const int data_idx, const bool is_tail,
H A Djit_avx512_core_x8s8s32x_1x1_conv_kernel.hpp117 void apply_postops(const int load_loop_blk, const int ur,
H A Djit_gemm_x8s8s32x_convolution_utils.cpp49 void apply_postops(const Xbyak::Reg64 &reg_dst, const int idx);
327 void jit_pp_ker_t::apply_postops(const Xbyak::Reg64 &reg_dst, const int idx) { in apply_postops() function in dnnl::impl::cpu::x64::gemm_x8s8s32x_convolution_utils::jit_pp_ker_t
476 apply_postops(reg_dst_, idx); in generate()

1234