| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_inner_product.hpp | 45 const auto wei_type = weights_md(0)->data_type; in init() local
52 && platform::has_data_type_support(wei_type) in init()
56 && utils::one_of(wei_type, f32, bf16) in init()
58 && src_type == wei_type in init()
99 const auto wei_type = weights_md(0)->data_type; in init() local
106 && platform::has_data_type_support(wei_type) in init()
109 && utils::one_of(wei_type, f32, bf16) in init()
111 && diff_dst_type == wei_type in init()
|
| H A D | ref_convolution.hpp | 44 const auto wei_type = weights_md(0)->data_type; in init() local
51 && platform::has_data_type_support(wei_type) in init()
55 && utils::one_of(wei_type, f32, bf16) in init()
57 && src_type == wei_type in init()
115 const auto wei_type = weights_md(0)->data_type; in init() local
121 && platform::has_data_type_support(wei_type) in init()
124 && utils::one_of(wei_type, f32, bf16) in init()
126 && diff_dst_type == wei_type in init()
|
| H A D | ref_convolution_int8.hpp | 44 const auto wei_type = weights_md(0)->data_type; in init() local
50 && utils::one_of(src_type, s8, u8) && wei_type == s8 in init()
126 const auto wei_type = weights_md(0)->data_type; in init() local
131 && utils::one_of(diff_dst_type, s8, u8) && wei_type == s8 in init()
|
| H A D | ref_inner_product_int8.hpp | 44 const auto wei_type = weights_md(0)->data_type; in init() local
51 && wei_type == s8 in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_inner_product.hpp | 35 template <data_type_t src_type, data_type_t wei_type = src_type,
48 && expect_data_types(src_type, wei_type, data_type::undef, in init()
51 && platform::has_data_type_support(wei_type) in init()
86 typedef typename prec_traits<wei_type>::type wei_data_t;
101 template <data_type_t diff_src_type, data_type_t wei_type,
112 && weights_md()->data_type == wei_type in init()
124 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | ref_convolution.hpp | 35 template <data_type_t src_type, data_type_t wei_type = src_type,
49 && expect_data_types(src_type, wei_type, data_type::undef, in init()
52 && platform::has_data_type_support(wei_type) in init()
116 typedef typename prec_traits<wei_type>::type wei_data_t;
130 template <data_type_t diff_src_type, data_type_t wei_type,
141 && expect_data_types(diff_src_type, wei_type, in init()
144 && platform::has_data_type_support(wei_type) in init()
180 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_brgemm_conv.cpp | 61 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
154 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
156 brgemm_convolution_fwd_t<isa, src_type, wei_type,
160 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
189 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
220 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
261 void brgemm_convolution_fwd_t<isa, src_type, wei_type,
340 last_ic_block = (wei_type == f32) ? 1 : ((wei_type == bf16) ? 2 : 4); in init()
600 void brgemm_convolution_fwd_t<isa, src_type, wei_type,
663 void brgemm_convolution_fwd_t<isa, src_type, wei_type,
[all …]
|
| H A D | jit_brgemm_1x1_conv.cpp | 40 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
43 brgemm_1x1_convolution_fwd_t<isa, src_type, wei_type, dst_type>::pd_t::init( in init()
58 && expect_data_types(src_type, wei_type, data_type::undef, dst_type, in init()
103 CHECK(brgemm_desc_init(&brg, isa, jcp_.brg_type, src_type, wei_type, in init()
125 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
127 status_t brgemm_1x1_convolution_fwd_t<isa, src_type, wei_type, dst_type>::init( in init()
172 || (one_of(src_type, u8, s8) && wei_type == s8) // oscales needed in init()
196 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
198 void brgemm_1x1_convolution_fwd_t<isa, src_type, wei_type, dst_type>::exec_ker( in exec_ker()
300 template <cpu_isa_t isa, data_type_t src_type, data_type_t wei_type,
[all …]
|
| H A D | jit_avx512_common_convolution.hpp | 38 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
52 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
77 typedef typename prec_traits<wei_type>::type wei_data_t;
113 impl::data_type_t wei_type = diff_dst_type,
127 && expect_data_types(diff_src_type, wei_type, in init()
152 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | jit_avx512_common_1x1_convolution.hpp | 42 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
63 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
244 typedef typename prec_traits<wei_type>::type wei_data_t;
290 impl::data_type_t wei_type = diff_dst_type,
306 && expect_data_types(diff_src_type, wei_type, in init()
354 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | jit_brgemm_1x1_conv.hpp | 41 impl::data_type_t wei_type = src_type,
71 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | jit_avx512_core_amx_1x1_convolution.cpp | 48 template <data_type_t src_type, data_type_t wei_type, data_type_t dst_type>
49 void jit_avx512_core_amx_1x1_convolution_fwd_t<src_type, wei_type,
64 template <data_type_t src_type, data_type_t wei_type, data_type_t dst_type>
65 status_t jit_avx512_core_amx_1x1_convolution_fwd_t<src_type, wei_type,
|
| H A D | jit_avx512_core_amx_convolution.hpp | 39 template <impl::data_type_t src_type, impl::data_type_t wei_type,
108 typedef typename prec_traits<wei_type>::type wei_data_t;
137 template <impl::data_type_t diff_src_type, impl::data_type_t wei_type,
180 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_sve_512_convolution.hpp | 38 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
52 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
76 typedef typename prec_traits<wei_type>::type wei_data_t;
112 impl::data_type_t wei_type = diff_dst_type,
126 && expect_data_types(diff_src_type, wei_type, in init()
149 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | jit_sve_512_1x1_convolution.hpp | 40 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
61 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
119 typedef typename prec_traits<wei_type>::type wei_data_t;
154 impl::data_type_t wei_type = diff_dst_type,
169 && expect_data_types(diff_src_type, wei_type, in init()
224 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | acl_gemm_convolution.cpp | 36 template <data_type_t src_type, data_type_t wei_type, data_type_t dst_type,
38 status_t acl_gemm_convolution_fwd_t<src_type, wei_type, dst_type,
|
| H A D | acl_gemm_convolution.hpp | 87 template <data_type_t src_type, data_type_t wei_type = src_type,
105 src_type, wei_type, bia_type, dst_type, undef) in init()
194 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_sve_512_convolution.hpp | 38 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
52 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
76 typedef typename prec_traits<wei_type>::type wei_data_t;
112 impl::data_type_t wei_type = diff_dst_type,
126 && expect_data_types(diff_src_type, wei_type, in init()
149 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | acl_gemm_convolution.cpp | 26 template <data_type_t src_type, data_type_t wei_type, data_type_t dst_type,
28 status_t acl_gemm_convolution_fwd_t<src_type, wei_type, dst_type,
|
| H A D | jit_sve_512_1x1_convolution.hpp | 40 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
61 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
119 typedef typename prec_traits<wei_type>::type wei_data_t;
154 impl::data_type_t wei_type = diff_dst_type,
169 && expect_data_types(diff_src_type, wei_type, in init()
224 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | acl_gemm_convolution.hpp | 80 template <data_type_t src_type, data_type_t wei_type = src_type,
98 src_type, wei_type, bia_type, dst_type, undef) in init()
173 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx512_common_convolution.hpp | 38 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
52 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
76 typedef typename prec_traits<wei_type>::type wei_data_t;
112 impl::data_type_t wei_type = diff_dst_type,
126 && expect_data_types(diff_src_type, wei_type, in init()
151 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| H A D | jit_avx512_common_1x1_convolution.hpp | 42 template <impl::data_type_t src_type, impl::data_type_t wei_type = src_type,
63 && expect_data_types(src_type, wei_type, dst_type, dst_type, in init()
240 typedef typename prec_traits<wei_type>::type wei_data_t;
286 impl::data_type_t wei_type = diff_dst_type,
302 && expect_data_types(diff_src_type, wei_type, in init()
350 typedef typename prec_traits<wei_type>::type wei_data_t;
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/matmul/ |
| H A D | ref_matmul.hpp | 48 const auto wei_type = weights_md(0)->data_type; in init() local
53 && utils::one_of(wei_type, f32, bf16) in init()
55 && src_type == wei_type in init()
|
| H A D | ref_matmul_int8.hpp | 47 const auto wei_type = weights_md(0)->data_type; in init() local
51 bool ok = utils::one_of(src_type, s8, u8) && wei_type == s8 in init()
|