| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_resampling_kernel.cpp | 239 const Vmm vmm_dst(data_idx); in apply_sum() local
510 const Vmm vmm_dst(vmm_idx(0)); in linear_ncsp_format() local
|
| H A D | jit_uni_layer_normalization_kernels.cpp | 102 Ymm &vmm_dst, Reg64 reg_dst, int nelems, size_t offt_elems) { in store()
229 Vmm vmm_dst = vmm_data; member
318 compute([&](Vmm vmm_dst) { vaddps(vmm_dst, vmm_dst, vmm_src); }); in generate()
324 compute([&](Vmm vmm_dst) { in generate()
|
| H A D | jit_uni_x8s8s32x_deconvolution.cpp | 623 const auto vmm_dst = vmm_out(it_ow, ocb); in append_zp_src_pad_str_comp() local
752 const auto vmm_dst = vmm_out(jj, ocb); in compute_ker() local
1081 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
1131 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 450 Vmm vmm_dst in store_dst() local
1064 inline void jit_uni_dw_conv_bwd_weights_kernel_f32<isa>::addps_xmm(Vmm &vmm_dst, in addps_xmm()
1071 Vmm &vmm_dst, Vmm &vmm_src, const Xbyak::Address &addr, in addps_xmm()
1083 Vmm &vmm_dst, Vmm &vmm_src, const Xbyak::Address &addr, in addps_xmm()
|
| H A D | jit_gemm_inner_product_utils.cpp | 1000 auto vmm_dst = vreg_dst(0); in compute_mb_blk() local
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 151 Vmm _jit_avx512_core_bf16_fwd_kernel<Vmm>::vmm_dst(int i_ur, int i_oc) const { in vmm_dst() function in dnnl::impl::cpu::x64::_jit_avx512_core_bf16_fwd_kernel
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_resampling_kernel.cpp | 239 const Vmm vmm_dst(data_idx); in apply_sum() local
495 const Vmm vmm_dst(vmm_idx(0)); in linear_ncsp_format() local
|
| H A D | jit_uni_layer_normalization_kernels.cpp | 102 Ymm &vmm_dst, Reg64 reg_dst, int nelems, size_t offt_elems) { in store()
229 Vmm vmm_dst = vmm_data; member
318 compute([&](Vmm vmm_dst) { vaddps(vmm_dst, vmm_dst, vmm_src); }); in generate()
324 compute([&](Vmm vmm_dst) { in generate()
|
| H A D | jit_avx512_core_x8s8s32x_deconvolution.cpp | 522 const auto vmm_dst = vmm_out(it_ow, ocb); in append_zp_src_pad_str_comp() local
653 const auto vmm_dst = vmm_out(jj, ocb); in compute_ker() local
942 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
1048 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
|
| H A D | jit_uni_x8s8s32x_deconvolution.cpp | 625 const auto vmm_dst = vmm_out(it_ow, ocb); in append_zp_src_pad_str_comp() local
754 const auto vmm_dst = vmm_out(jj, ocb); in compute_ker() local
1093 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
1144 const auto vmm_dst = vmm_out(ur, ocb); in store_output() local
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 423 Vmm vmm_dst in store_dst() local
1037 inline void jit_uni_dw_conv_bwd_weights_kernel_f32<isa>::addps_xmm(Vmm &vmm_dst, in addps_xmm()
1044 Vmm &vmm_dst, Vmm &vmm_src, const Xbyak::Address &addr, in addps_xmm()
1056 Vmm &vmm_dst, Vmm &vmm_src, const Xbyak::Address &addr, in addps_xmm()
|
| H A D | jit_gemm_inner_product_utils.cpp | 987 auto vmm_dst = vreg_dst(0); in compute_mb_blk() local
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 151 Vmm _jit_avx512_core_bf16_fwd_kernel<Vmm>::vmm_dst(int i_ur, int i_oc) const { in vmm_dst() function in dnnl::impl::cpu::x64::_jit_avx512_core_bf16_fwd_kernel
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/injectors/ |
| H A D | jit_uni_eltwise_injector.cpp | 189 void jit_uni_eltwise_injector_f32<isa>::vec_shift(const Vmm &vmm_dst, in vec_shift()
230 const Vmm &vmm_dst, const Xbyak::Operand &src) { in blend_with_mask()
358 Vmm vmm_dst = vmm_aux1, vmm_src_shift = vmm_aux1, vmm_coeff = vmm_aux1, in tanh_compute_vector_fwd() local
894 size_t offt = 0) { in log_compute_vector_fwd()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/injectors/ |
| H A D | jit_uni_eltwise_injector.cpp | 189 void jit_uni_eltwise_injector_f32<isa, Wmm>::vec_shift(const Vmm &vmm_dst, in vec_shift()
231 const Vmm &vmm_dst, const Xbyak::Operand &src) { in blend_with_mask()
359 Vmm vmm_dst = vmm_aux1, vmm_src_shift = vmm_aux1, vmm_coeff = vmm_aux1, in tanh_compute_vector_fwd() local
897 size_t offt = 0) { in log_compute_vector_fwd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/injectors/ |
| H A D | jit_uni_eltwise_injector.cpp | 436 const TRegS &vmm_dst, const TRegS &src) { in blend_with_mask()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/injectors/ |
| H A D | jit_uni_eltwise_injector.cpp | 436 const TRegS &vmm_dst, const TRegS &src) { in blend_with_mask()
|