Searched refs:weights_diff_acc_vmm_ (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/prelu/ |
| H A D | jit_uni_prelu_backward_kernel.cpp | 106 , weights_diff_acc_vmm_( in jit_uni_prelu_backward_kernel_t()
139 vmovups(weights_diff_acc_vmm_, ptr[reg_weights_diff_]); in prepare_kernel_const_vars()
142 uni_vxorps(weights_diff_acc_vmm_, weights_diff_acc_vmm_, in prepare_kernel_const_vars()
143 weights_diff_acc_vmm_); in prepare_kernel_const_vars()
144 uni_vmovss(weights_diff_acc_vmm_, ptr[reg_weights_diff_]); in prepare_kernel_const_vars()
248 accumulate_weights_diff(weights_diff_vmm, weights_diff_acc_vmm_, in compute_dst()
278 weights_diff_acc_vmm_, weights_diff_acc_vmm_, partial_sum_vmm); in accumulate_weights_diff()
342 uni_vmovups(ptr[reg_weights_diff_], weights_diff_acc_vmm_); in finalize()
344 reduce(this, weights_diff_acc_vmm_, weights_const_vmm_, isa_); in finalize()
345 uni_vmovss(ptr[reg_weights_diff_], weights_diff_acc_vmm_); in finalize()
|
| H A D | jit_uni_prelu_backward_kernel.hpp | 104 const Vmm weights_diff_acc_vmm_; member in dnnl::impl::cpu::x64::jit_uni_prelu_backward_kernel_t
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/prelu/ |
| H A D | jit_uni_prelu_backward_kernel.cpp | 106 , weights_diff_acc_vmm_( in jit_uni_prelu_backward_kernel_t()
139 vmovups(weights_diff_acc_vmm_, ptr[reg_weights_diff_]); in prepare_kernel_const_vars()
142 uni_vxorps(weights_diff_acc_vmm_, weights_diff_acc_vmm_, in prepare_kernel_const_vars()
143 weights_diff_acc_vmm_); in prepare_kernel_const_vars()
144 uni_vmovss(weights_diff_acc_vmm_, ptr[reg_weights_diff_]); in prepare_kernel_const_vars()
248 accumulate_weights_diff(weights_diff_vmm, weights_diff_acc_vmm_, in compute_dst()
278 weights_diff_acc_vmm_, weights_diff_acc_vmm_, partial_sum_vmm); in accumulate_weights_diff()
342 uni_vmovups(ptr[reg_weights_diff_], weights_diff_acc_vmm_); in finalize()
344 reduce(this, weights_diff_acc_vmm_, weights_const_vmm_, isa_); in finalize()
345 uni_vmovss(ptr[reg_weights_diff_], weights_diff_acc_vmm_); in finalize()
|
| H A D | jit_uni_prelu_backward_kernel.hpp | 104 const Vmm weights_diff_acc_vmm_; member in dnnl::impl::cpu::x64::jit_uni_prelu_backward_kernel_t
|