Home
last modified time | relevance | path

Searched defs:wei_type (Results 1 – 13 of 13) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dref_inner_product.hpp45 const auto wei_type = weights_md(0)->data_type; in init() local
99 const auto wei_type = weights_md(0)->data_type; in init() local
H A Dref_convolution_int8.hpp44 const auto wei_type = weights_md(0)->data_type; in init() local
126 const auto wei_type = weights_md(0)->data_type; in init() local
H A Dref_convolution.hpp44 const auto wei_type = weights_md(0)->data_type; in init() local
115 const auto wei_type = weights_md(0)->data_type; in init() local
H A Dref_inner_product_int8.hpp44 const auto wei_type = weights_md(0)->data_type; in init() local
H A Dref_deconvolution.hpp336 auto wei_type = desc()->weights_desc.data_type; in init() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/matmul/
H A Dref_matmul.hpp48 const auto wei_type = weights_md(0)->data_type; in init() local
H A Dref_matmul_int8.hpp47 const auto wei_type = weights_md(0)->data_type; in init() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_brgemm_1x1_conv.cpp48 const auto wei_type = weights_md(0)->data_type; in init() local
161 const auto wei_type = pd()->weights_md(0)->data_type; in init() local
H A Djit_brdgmm_dw_conv.cpp76 const auto wei_type = cd.weights_desc.data_type; in init() local
H A Djit_brgemm_conv.cpp49 const auto wei_type = weights_md(0)->data_type; in init() local
413 const auto wei_type = pd()->weights_md(0)->data_type; in init() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dref_deconvolution.hpp341 auto wei_type = desc()->weights_desc.data_type; in init() local
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/
H A Dxe_hp_conv_bwd_wei_kernel.cpp313 DataType src_type, wei_type, bia_type, dst_type; member in dnnl::impl::gpu::jit::xe_hp_conv_bwd_wei_conv_kernel_t
H A Dxe_hp_conv_data_kernel.cpp2978 DataType wei_type; member in dnnl::impl::gpu::jit::__anon3ab3df010111::xe_hp_conv_data_kernel_t