| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_layer_normalization.cpp | 435 const memory::desc diff_src_d = diff_src.get_desc(); in check_lnorm_bwd() local
447 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in check_lnorm_bwd()
|
| H A D | test_batch_normalization_common.hpp | 542 const memory::desc diff_src_d = diff_src.get_desc(); in check_bnorm_bwd() local
553 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in check_bnorm_bwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_layer_normalization.cpp | 435 const memory::desc diff_src_d = diff_src.get_desc(); in check_lnorm_bwd() local
447 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in check_lnorm_bwd()
|
| H A D | test_batch_normalization_common.hpp | 542 const memory::desc diff_src_d = diff_src.get_desc(); in check_bnorm_bwd() local
553 const dnnl::impl::memory_desc_wrapper diff_src_mdw(diff_src_d.data); in check_bnorm_bwd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_common_1x1_convolution.cpp | 451 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
459 const int ndims = diff_src_d.ndims(); in execute_backward_data()
543 diff_src_d, n, ic_off_idx, id, ih, iw); in execute_backward_data()
|
| H A D | jit_avx2_conv_kernel_f32.hpp | 147 const convolution_desc_t &cd, const memory_desc_wrapper &diff_src_d,
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 3461 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
3466 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
3467 int ndims = diff_src_d.ndims(); in init_conf()
3478 one_of(diff_src_d.data_type(), bf16, f32)); in init_conf()
3482 one_of(diff_src_d.data_type(), f32, s32, s8, u8)); in init_conf()
3496 jcp.mb = diff_src_d.dims()[0]; in init_conf()
3499 jcp.ic = diff_src_d.dims()[1] / jcp.ngroups; in init_conf()
3501 jcp.ih = !is_1d ? diff_src_d.dims()[ndims - 2] : 1; in init_conf()
3502 jcp.iw = diff_src_d.dims()[ndims - 1]; in init_conf()
3554 if (diff_src_d.format_kind() == format_kind::any) { in init_conf()
[all …]
|
| H A D | jit_avx2_1x1_convolution.cpp | 375 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
448 + data_blk_off(diff_src_d, n, ic_off_idx, id, ih, iw); in execute_backward_data()
|
| H A D | jit_avx512_common_conv_kernel.cpp | 2767 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
2773 int ndims = diff_src_d.ndims(); in init_conf()
2780 jcp.mb = diff_src_d.dims()[0]; in init_conf()
2784 jcp.ic = diff_src_d.dims()[1] / jcp.ngroups; in init_conf()
2787 jcp.id = (ndims == 5) ? diff_src_d.dims()[2] : 1; in init_conf()
2788 jcp.ih = (ndims == 3) ? 1 : diff_src_d.dims()[ndims - 2]; in init_conf()
2789 jcp.iw = diff_src_d.dims()[ndims - 1]; in init_conf()
2833 auto curr_src_tag = diff_src_d.matches_one_of_tag( in init_conf()
2844 && diff_src_d.data_type() == data_type::f32; in init_conf()
2943 && diff_src_d.data_type() == data_type::f32) { in init_conf()
[all …]
|
| H A D | jit_uni_dw_convolution.cpp | 177 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
202 par_conv.src = &diff_src[diff_src_d.blk_off(n, ic_offset, ih, iw)]; in execute_backward_data()
|
| H A D | jit_avx512_common_conv_kernel.hpp | 384 const convolution_desc_t &cd, memory_desc_t &diff_src_d,
|
| H A D | jit_avx512_common_conv_winograd_kernel_f32.cpp | 588 const memory_desc_wrapper &diff_src_d, in init_conf() argument
591 status_t st = init_conf_common(jcp, cd, diff_src_d, weights_d, diff_dst_d); in init_conf()
|
| H A D | jit_uni_batch_normalization.cpp | 2138 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
2144 = diff_src_d.matches_one_of_tag(nChw16c, nCdhw16c, nhwc, ndhwc); in init()
2147 diff_src_tag = diff_src_d.matches_one_of_tag(nChw8c, nCdhw8c); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx512_common_1x1_convolution.cpp | 452 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
460 const int ndims = diff_src_d.ndims(); in execute_backward_data()
544 diff_src_d, n, ic_off_idx, id, ih, iw); in execute_backward_data()
|
| H A D | jit_avx2_conv_kernel_f32.hpp | 150 const convolution_desc_t &cd, const memory_desc_wrapper &diff_src_d,
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 3484 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
3489 const bool with_groups = weights_d.ndims() == diff_src_d.ndims() + 1; in init_conf()
3490 int ndims = diff_src_d.ndims(); in init_conf()
3500 one_of(diff_src_d.data_type(), bf16, f32)); in init_conf()
3506 one_of(diff_src_d.data_type(), f32, s32, s8, u8)); in init_conf()
3518 jcp.mb = diff_src_d.dims()[0]; in init_conf()
3521 jcp.ic = diff_src_d.dims()[1] / jcp.ngroups; in init_conf()
3523 jcp.ih = !is_1d ? diff_src_d.dims()[ndims - 2] : 1; in init_conf()
3524 jcp.iw = diff_src_d.dims()[ndims - 1]; in init_conf()
3575 if (diff_src_d.format_kind() == format_kind::any) { in init_conf()
[all …]
|
| H A D | jit_avx2_1x1_convolution.cpp | 375 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
448 + data_blk_off(diff_src_d, n, ic_off_idx, id, ih, iw); in execute_backward_data()
|
| H A D | jit_avx512_common_conv_kernel.cpp | 2755 const memory_desc_wrapper diff_src_d(&diff_src_md); in init_conf() local
2761 int ndims = diff_src_d.ndims(); in init_conf()
2768 jcp.mb = diff_src_d.dims()[0]; in init_conf()
2772 jcp.ic = diff_src_d.dims()[1] / jcp.ngroups; in init_conf()
2775 jcp.id = (ndims == 5) ? diff_src_d.dims()[2] : 1; in init_conf()
2776 jcp.ih = (ndims == 3) ? 1 : diff_src_d.dims()[ndims - 2]; in init_conf()
2777 jcp.iw = diff_src_d.dims()[ndims - 1]; in init_conf()
2821 auto curr_src_tag = diff_src_d.matches_one_of_tag( in init_conf()
2832 && diff_src_d.data_type() == data_type::f32; in init_conf()
2931 && diff_src_d.data_type() == data_type::f32) { in init_conf()
[all …]
|
| H A D | jit_uni_dw_convolution.cpp | 179 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
204 par_conv.src = &diff_src[diff_src_d.blk_off(n, ic_offset, ih, iw)]; in execute_backward_data()
|
| H A D | jit_avx512_common_conv_kernel.hpp | 388 const convolution_desc_t &cd, memory_desc_t &diff_src_d,
|
| H A D | jit_avx512_common_conv_winograd_kernel_f32.cpp | 588 const memory_desc_wrapper &diff_src_d, in init_conf() argument
591 status_t st = init_conf_common(jcp, cd, diff_src_d, weights_d, diff_dst_d); in init_conf()
|
| H A D | jit_uni_batch_normalization.cpp | 2137 const memory_desc_wrapper diff_src_d(diff_src_md()); in init() local
2143 = diff_src_d.matches_one_of_tag(nChw16c, nCdhw16c, nhwc, ndhwc); in init()
2146 diff_src_tag = diff_src_d.matches_one_of_tag(nChw8c, nCdhw8c); in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_convolution.cpp | 290 const memory_desc_wrapper diff_src_d(pd()->diff_src_md()); in execute_backward_data() local
455 diff_src_d, ndims, mb, g * IC + ic, id, ih, iw); in execute_backward_data()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_sve_512_conv_kernel.hpp | 237 const convolution_desc_t &cd, memory_desc_t &diff_src_d,
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_sve_512_conv_kernel.hpp | 237 const convolution_desc_t &cd, memory_desc_t &diff_src_d,
|