Searched refs:os_nb (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | gemm_convolution.cpp | 556 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_data_ncsp() local
557 auto out_off = os_nb * m + od * jcp.os; in execute_backward_data_ncsp()
561 = nstl::min((dim_t)jcp.os_block, jcp.os - os_nb * m); in execute_backward_data_ncsp()
576 os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
579 _diff_src, od, os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
824 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_weights_ncsp() local
825 auto out_off = os_nb * k + od * jcp.os; in execute_backward_weights_ncsp()
827 (dim_t)jcp.os_block, jcp.os - os_nb * k); in execute_backward_weights_ncsp()
834 _src, _col, os_nb * jcp.os_block, in execute_backward_weights_ncsp()
839 os_nb * jcp.os_block, os_block); in execute_backward_weights_ncsp()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | gemm_convolution.cpp | 553 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_data_ncsp() local
554 auto out_off = os_nb * m + od * jcp.os; in execute_backward_data_ncsp()
558 = nstl::min((dim_t)jcp.os_block, jcp.os - os_nb * m); in execute_backward_data_ncsp()
573 os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
576 _diff_src, od, os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
821 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_weights_ncsp() local
822 auto out_off = os_nb * k + od * jcp.os; in execute_backward_weights_ncsp()
824 (dim_t)jcp.os_block, jcp.os - os_nb * k); in execute_backward_weights_ncsp()
831 _src, _col, os_nb * jcp.os_block, in execute_backward_weights_ncsp()
836 os_nb * jcp.os_block, os_block); in execute_backward_weights_ncsp()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | gemm_bf16_convolution.cpp | 847 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_data_ncsp() local
848 auto out_off = os_nb * m + od * jcp.os; in execute_backward_data_ncsp()
852 = nstl::min((dim_t)jcp.os_block, jcp.os - os_nb * m); in execute_backward_data_ncsp()
871 od, os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
1249 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_weights_ncsp() local
1250 auto out_off = os_nb * k + od * jcp.os; in execute_backward_weights_ncsp()
1252 (dim_t)jcp.os_block, jcp.os - os_nb * k); in execute_backward_weights_ncsp()
1259 jcp, _src, _col, os_nb * jcp.os_block, in execute_backward_weights_ncsp()
1264 os_nb * jcp.os_block, os_block); in execute_backward_weights_ncsp()
1273 mb == mb_start && os_nb == 0 && od == 0 ? &zero in execute_backward_weights_ncsp()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | gemm_bf16_convolution.cpp | 839 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_data_ncsp() local
840 auto out_off = os_nb * m + od * jcp.os; in execute_backward_data_ncsp()
844 = nstl::min((dim_t)jcp.os_block, jcp.os - os_nb * m); in execute_backward_data_ncsp()
863 od, os_nb * jcp.os_block, os_block); in execute_backward_data_ncsp()
1240 for (int os_nb = 0; os_nb < jcp.os_nb_block; ++os_nb) { in execute_backward_weights_ncsp() local
1241 auto out_off = os_nb * k + od * jcp.os; in execute_backward_weights_ncsp()
1243 (dim_t)jcp.os_block, jcp.os - os_nb * k); in execute_backward_weights_ncsp()
1250 jcp, _src, _col, os_nb * jcp.os_block, in execute_backward_weights_ncsp()
1255 os_nb * jcp.os_block, os_block); in execute_backward_weights_ncsp()
1264 mb == mb_start && os_nb == 0 && od == 0 ? &zero in execute_backward_weights_ncsp()
[all …]
|