Searched refs:diff_src_dt_ (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 | 35 , diff_src_dt_(pd->diff_src_md(0)->data_type) in jit_prelu_backward_kernel_t() 78 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, diff_src_dt_, in any_tensor_bf16() 88 diff_src_dt_, data_type::u8, data_type::s8, data_type::s32)) in jit_uni_prelu_backward_kernel_t() 95 ? (diff_wei_dt_ == diff_src_dt_ in jit_uni_prelu_backward_kernel_t() 128 io_.init_saturate_f32({diff_src_dt_, diff_wei_dt_}); in prepare_kernel_const_vars() 193 io_.at(diff_src_dt_) in compute_dst() 197 prelu::apply_zero_padding(this, tail_size_, diff_src_dt_, in compute_dst() 261 io_.at(diff_src_dt_) in compute_dst() 265 prelu::apply_zero_padding(this, tail_size_, diff_src_dt_, in compute_dst() 356 saturation_map.emplace(diff_src_dt_, in create_saturation_vmm_map() [all …]
|
H A D | jit_uni_prelu_backward_kernel.hpp | 61 const data_type_t diff_src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/prelu/ |
H A D | jit_uni_prelu_backward_kernel.cpp | 35 , diff_src_dt_(pd->diff_src_md(0)->data_type) in jit_prelu_backward_kernel_t() 78 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, diff_src_dt_, in any_tensor_bf16() 88 diff_src_dt_, data_type::u8, data_type::s8, data_type::s32)) in jit_uni_prelu_backward_kernel_t() 95 ? (diff_wei_dt_ == diff_src_dt_ in jit_uni_prelu_backward_kernel_t() 128 io_.init_saturate_f32({diff_src_dt_, diff_wei_dt_}); in prepare_kernel_const_vars() 193 io_.at(diff_src_dt_) in compute_dst() 197 prelu::apply_zero_padding(this, tail_size_, diff_src_dt_, in compute_dst() 261 io_.at(diff_src_dt_) in compute_dst() 265 prelu::apply_zero_padding(this, tail_size_, diff_src_dt_, in compute_dst() 356 saturation_map.emplace(diff_src_dt_, in create_saturation_vmm_map() [all …]
|
H A D | jit_uni_prelu_backward_kernel.hpp | 61 const data_type_t diff_src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t
|