| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_inner_product.cpp | 147 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
154 = utils::one_of(diff_src_d.ndims(), 3, 4, 5); in execute_backward_data()
155 const auto ndims = diff_src_d.ndims() - 2; in execute_backward_data()
186 diff_src[diff_src_d.off(mb, ic, kd, kh, kw)] in execute_backward_data()
190 diff_src[diff_src_d.off(mb, ic, kh, kw)] in execute_backward_data()
194 diff_src[diff_src_d.off(mb, ic, kw)] in execute_backward_data()
206 diff_src[diff_src_d.off(mb, ic)] = (diff_src_data_t)ds; in execute_backward_data()
|
| H A D | ref_pooling.cpp | 221 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward() local
246 const auto off = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
272 const auto off = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
316 const auto off = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
|
| H A D | ref_deconvolution.hpp | 211 const memory_desc_wrapper diff_src_d(conv_pd_->diff_src_md()); in init_scratchpad() local
212 assert(diff_src_d.data_type_size() == sizeof(float)); in init_scratchpad()
213 scratchpad.book(key_deconv_bias, diff_src_d.nelems(true), in init_scratchpad()
214 diff_src_d.data_type_size()); in init_scratchpad()
|
| H A D | ref_resampling.cpp | 214 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward() local
258 get_offset(diff_src_d, mb, ch, id, ih, iw)); in execute_backward()
283 get_offset(diff_src_d, mb, ch, id, ih, iw)); in execute_backward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_dw_conv_kernel_utils.hpp | 274 const convolution_desc_t &cd, const memory_desc_wrapper &diff_src_d,
292 const memory_desc_wrapper &diff_src_d, in init_conf() argument
305 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
309 jcp.mb = diff_src_d.dims()[0]; in init_conf()
313 jcp.ic = diff_src_d.dims()[1]; in init_conf()
315 jcp.ih = diff_src_d.dims()[2]; in init_conf()
316 jcp.iw = diff_src_d.dims()[3]; in init_conf()
358 jcp.src_tag = diff_src_d.matches_one_of_tag(dat_tag); in init_conf()
365 && jcp.ic <= diff_src_d.padded_dims()[1] in init_conf()
370 jcp.typesize_out = types::data_type_size(diff_src_d.data_type()); in init_conf()
|
| H A D | jit_uni_softmax.hpp | 109 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
133 bool ok = dst_d == diff_dst_d && dst_d == diff_src_d && mayiuse(isa) in init()
|
| H A D | jit_sve_512_1x1_convolution.hpp | 176 const memory_desc_t *diff_src_d = diff_src_md(); in init() local
177 rtus_prepare(this, conv_d, diff_src_d, diff_dst_md()); in init()
180 *conv_d, *diff_src_d, *weights_md(), *diff_dst_md(), in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_dw_conv_kernel_utils.hpp | 274 const convolution_desc_t &cd, const memory_desc_wrapper &diff_src_d,
292 const memory_desc_wrapper &diff_src_d, in init_conf() argument
305 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
309 jcp.mb = diff_src_d.dims()[0]; in init_conf()
313 jcp.ic = diff_src_d.dims()[1]; in init_conf()
315 jcp.ih = diff_src_d.dims()[2]; in init_conf()
316 jcp.iw = diff_src_d.dims()[3]; in init_conf()
358 jcp.src_tag = diff_src_d.matches_one_of_tag(dat_tag); in init_conf()
365 && jcp.ic <= diff_src_d.padded_dims()[1] in init_conf()
370 jcp.typesize_out = types::data_type_size(diff_src_d.data_type()); in init_conf()
|
| H A D | jit_uni_softmax.hpp | 109 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
133 bool ok = dst_d == diff_dst_d && dst_d == diff_src_d && mayiuse(isa) in init()
|
| H A D | jit_sve_512_1x1_convolution.hpp | 176 const memory_desc_t *diff_src_d = diff_src_md(); in init() local
177 rtus_prepare(this, conv_d, diff_src_d, diff_dst_md()); in init()
180 *conv_d, *diff_src_d, *weights_md(), *diff_dst_md(), in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_core_amx_conv_utils.hpp | 41 const memory_desc_wrapper &diff_src_d) { in execute_backward_convolution_body() argument
45 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in execute_backward_convolution_body()
167 ? diff_src_d.blk_off(mb, icb, iw) in execute_backward_convolution_body()
168 : diff_src_d.blk_off(mb, icb, ih, iw); in execute_backward_convolution_body()
|
| H A D | jit_uni_dw_conv_kernel_utils.hpp | 328 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
339 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
342 const int ndims = diff_src_d.ndims(); in init_conf()
344 jcp.mb = diff_src_d.dims()[0]; in init_conf()
348 jcp.ic = diff_src_d.dims()[1]; in init_conf()
350 jcp.ih = diff_src_d.dims()[2]; in init_conf()
351 jcp.iw = diff_src_d.dims()[3]; in init_conf()
384 = diff_src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_blocked); in init_conf()
437 && jcp.ic <= diff_src_d.padded_dims()[1] in init_conf()
442 jcp.typesize_out = types::data_type_size(diff_src_d.data_type()); in init_conf()
|
| H A D | jit_uni_softmax.hpp | 116 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
140 bool ok = dst_d == diff_dst_d && dst_d == diff_src_d && mayiuse(isa) in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx512_core_amx_conv_utils.hpp | 41 const memory_desc_wrapper &diff_src_d) { in execute_backward_convolution_body() argument
45 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in execute_backward_convolution_body()
167 ? diff_src_d.blk_off(mb, icb, iw) in execute_backward_convolution_body()
168 : diff_src_d.blk_off(mb, icb, ih, iw); in execute_backward_convolution_body()
|
| H A D | jit_uni_dw_conv_kernel_utils.cpp | 248 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
259 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
262 const int ndims = diff_src_d.ndims(); in init_conf()
264 jcp.mb = diff_src_d.dims()[0]; in init_conf()
268 jcp.ic = diff_src_d.dims()[1]; in init_conf()
270 jcp.ih = diff_src_d.dims()[2]; in init_conf()
271 jcp.iw = diff_src_d.dims()[3]; in init_conf()
304 = diff_src_d.matches_one_of_tag(dat_tag_nxc, dat_tag_blocked); in init_conf()
357 && jcp.ic <= diff_src_d.padded_dims()[1] in init_conf()
362 jcp.typesize_out = types::data_type_size(diff_src_d.data_type()); in init_conf()
|
| H A D | jit_uni_softmax.hpp | 116 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
140 bool ok = dst_d == diff_dst_d && dst_d == diff_src_d && mayiuse(isa) in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_inner_product.cpp | 107 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
134 diff_src_d, ndims, mb, ic, kd, kh, kw); in execute_backward_data()
136 diff_src_d.data_type(), ds, diff_src, diff_src_off); in execute_backward_data()
|
| H A D | ref_deconvolution.hpp | 216 const memory_desc_wrapper diff_src_d(conv_pd_->diff_src_md()); in init_scratchpad() local
217 assert(diff_src_d.data_type_size() == sizeof(float)); in init_scratchpad()
218 scratchpad.book(key_deconv_bias, diff_src_d.nelems(true), in init_scratchpad()
219 diff_src_d.data_type_size()); in init_scratchpad()
|
| H A D | ref_pooling.cpp | 212 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward() local
241 const auto off = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
266 const auto d_src_off = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
311 = get_offset(diff_src_d, mb, oc, id, ih, iw); in execute_backward()
|
| H A D | ref_resampling.cpp | 214 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward() local
258 get_offset(diff_src_d, mb, ch, id, ih, iw)); in execute_backward()
283 get_offset(diff_src_d, mb, ch, id, ih, iw)); in execute_backward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_convolution_backward_data_common.hpp | 35 const memory::desc diff_src_d = diff_src.get_desc(); in compute_ref_conv_bwd_data() local
39 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_conv_bwd_data()
42 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in compute_ref_conv_bwd_data()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_convolution_backward_data_common.hpp | 35 const memory::desc diff_src_d = diff_src.get_desc(); in compute_ref_conv_bwd_data() local
39 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_conv_bwd_data()
42 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in compute_ref_conv_bwd_data()
|
| /dports/math/ideep/ideep-2.0.0-119-gb57539e/tests/ |
| H A D | test_ideep_common.hpp | 218 const auto *diff_src_d = diff_src.get_mkldnn_memory_desc_t(); in compute_ref_conv_bwd_data() local
454 const auto *diff_src_d = diff_src.get_mkldnn_memory_desc_t(); in check_lrn_bwd() local
512 ker(&ref_diff_src_ptr[map_index(diff_src_d, off(mb, c, h, w))], in check_lrn_bwd()
515 auto B = diff_src_ptr[map_index(diff_src_d, off(mb, c, h, w))]; in check_lrn_bwd()
746 const auto *diff_src_d = gradx.get_mkldnn_memory_desc_t(); in check_pool_bwd() local
827 diff_src_data[map_index(diff_src_d, iidx)], in check_pool_bwd()
981 const auto* diff_src_d = diff_src.get_mkldnn_memory_desc_t(); in check_bnrm_bwd() local
1042 data_t out_diff_src = diff_src_data[map_index(diff_src_d, sidx)]; in check_bnrm_bwd()
1136 const auto* diff_src_d = diff_src.get_mkldnn_memory_desc_t(); in compute_ref_inner_product_bwd_data() local
1148 data_t *ds = &diff_src_data[map_index(diff_src_d, dsidx)]; in compute_ref_inner_product_bwd_data()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | ref_resampling.hpp | 199 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in init() local
217 set_offsets(diff_src_d, off.src_off); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | ref_resampling.hpp | 200 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in init() local
218 set_offsets(diff_src_d, off.src_off); in init()
|