| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | ref_matmul.hpp | 48 src_dt_ = src_md()->data_type; in init()
62 && ((utils::one_of(src_dt_, u8, s8) in init()
69 f32, src_dt_, wei_dt_, dst_dt_) in init()
71 f16, src_dt_, wei_dt_, dst_dt_) in init()
73 bf16, src_dt_, wei_dt_) in init()
119 data_type_t src_dt_ = data_type::undef; member
163 def_data_type(kernel_ctx, pd()->src_dt_, "SRC"); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | ref_matmul.hpp | 48 src_dt_ = src_md()->data_type; in init()
61 && ((utils::one_of(src_dt_, u8, s8) in init()
68 f32, src_dt_, wei_dt_, dst_dt_) in init()
70 f16, src_dt_, wei_dt_) in init()
74 bf16, src_dt_, wei_dt_) in init()
122 data_type_t src_dt_ = data_type::undef; member
166 def_data_type(kernel_ctx, pd()->src_dt_, "SRC"); in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/prelu/ |
| H A D | jit_uni_prelu_forward_kernel.cpp | 32 , src_dt_(pd->src_md(0)->data_type) in jit_prelu_forward_kernel_t()
58 case DNNL_ARG_SRC: return get_addr(reg_src_, src_dt_); in data_ptr()
67 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, dst_dt_); in any_tensor_bf16()
88 , io_(this, isa, {src_dt_, wei_dt_, dst_dt_}, {}, in jit_uni_prelu_forward_kernel_t()
223 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
|
| H A D | jit_uni_prelu_backward_kernel.cpp | 33 , src_dt_(pd->src_md(0)->data_type) in jit_prelu_backward_kernel_t()
65 case DNNL_ARG_SRC: return get_addr(reg_src_, src_dt_); in data_ptr()
78 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, diff_src_dt_, in any_tensor_bf16()
112 {src_dt_, wei_dt_, diff_src_dt_, diff_dst_dt_, diff_wei_dt_}, {}, in jit_uni_prelu_backward_kernel_t()
175 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
234 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
|
| H A D | jit_uni_prelu_forward_kernel.hpp | 48 const data_type_t src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_forward_kernel_t
|
| H A D | jit_uni_prelu_backward_kernel.hpp | 59 const data_type_t 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_forward_kernel.cpp | 32 , src_dt_(pd->src_md(0)->data_type) in jit_prelu_forward_kernel_t()
58 case DNNL_ARG_SRC: return get_addr(reg_src_, src_dt_); in data_ptr()
67 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, dst_dt_); in any_tensor_bf16()
88 , io_(this, isa, {src_dt_, wei_dt_, dst_dt_}, {}, in jit_uni_prelu_forward_kernel_t()
223 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
|
| H A D | jit_uni_prelu_backward_kernel.cpp | 33 , src_dt_(pd->src_md(0)->data_type) in jit_prelu_backward_kernel_t()
65 case DNNL_ARG_SRC: return get_addr(reg_src_, src_dt_); in data_ptr()
78 return utils::one_of(data_type::bf16, src_dt_, wei_dt_, diff_src_dt_, in any_tensor_bf16()
112 {src_dt_, wei_dt_, diff_src_dt_, diff_dst_dt_, diff_wei_dt_}, {}, in jit_uni_prelu_backward_kernel_t()
175 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
234 io_.at(src_dt_)->load(data_ptr(DNNL_ARG_SRC, offset), src_vmm, tail); in compute_dst()
|
| H A D | jit_uni_prelu_forward_kernel.hpp | 48 const data_type_t src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_forward_kernel_t
|
| H A D | jit_uni_prelu_backward_kernel.hpp | 59 const data_type_t src_dt_; member in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t
|