Home
last modified time | relevance | path

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 Djit_uni_prelu_backward_kernel.cpp35 , 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 Djit_uni_prelu_backward_kernel.hpp61 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 Djit_uni_prelu_backward_kernel.cpp35 , 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 Djit_uni_prelu_backward_kernel.hpp61 const data_type_t diff_src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t