Searched refs:need_reduction (Results 1 – 7 of 7) sorted by relevance
627 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local628 const dim_t LDC = need_reduction ? jcp.oc : jcp.ngroups * jcp.oc; in execute_backward_weights_nspc()654 data_t *_diff_weights = need_reduction in execute_backward_weights_nspc()696 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_nspc()719 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local721 if (need_reduction && ithr_g != -1 && ithr_mb != -1) { in execute_backward_weights_nspc()794 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local817 data_t *_diff_weights = need_reduction in execute_backward_weights_ncsp()861 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_ncsp()886 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local[all …]
624 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local625 const dim_t LDC = need_reduction ? jcp.oc : jcp.ngroups * jcp.oc; in execute_backward_weights_nspc()651 data_t *_diff_weights = need_reduction in execute_backward_weights_nspc()693 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_nspc()716 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local718 if (need_reduction && ithr_g != -1 && ithr_mb != -1) { in execute_backward_weights_nspc()791 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local814 data_t *_diff_weights = need_reduction in execute_backward_weights_ncsp()858 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_ncsp()883 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local[all …]
1017 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local1053 : need_reduction ? weights_reduce in execute_backward_weights_nspc()1057 : need_reduction ? jcp.oc : jcp.ngroups * jcp.oc; in execute_backward_weights_nspc()1098 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_nspc()1124 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local1126 if (need_reduction && ithr_g != -1 && ithr_mb != -1) { in execute_backward_weights_nspc()1219 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local1242 acc_data_t *acc = need_reduction in execute_backward_weights_ncsp()1288 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_ncsp()1324 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local[all …]
1762 const bool need_reduction = nthr_NS > 1; in exec_fwd_step_stats() local1768 if (!need_reduction) return; in exec_fwd_step_stats()1785 acc_data_t *r_mean = need_reduction ? rbuf : mean; in exec_fwd_step_stats()1803 c.do_normalise = !need_reduction; in exec_fwd_step_stats()1811 acc_data_t *r_var = need_reduction ? rbuf : var; in exec_fwd_step_stats()1830 c.do_normalise = !need_reduction; in exec_fwd_step_stats()1927 const bool need_reduction = nthr_NS > 1; in exec_bwd_step_diff_ss() local1932 acc_data_t *const r_diff_gamma = need_reduction ? rbuf : diff_gamma; in exec_bwd_step_diff_ss()1934 = need_reduction ? rbuf + nthr_NS * size_C_stat : diff_beta; in exec_bwd_step_diff_ss()1937 if (!need_reduction) return; in exec_bwd_step_diff_ss()
1008 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local1044 : need_reduction ? weights_reduce in execute_backward_weights_nspc()1048 : need_reduction ? jcp.oc : jcp.ngroups * jcp.oc; in execute_backward_weights_nspc()1089 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_nspc()1115 const int need_reduction = nthr_mb != 1; in execute_backward_weights_nspc() local1117 if (need_reduction && ithr_g != -1 && ithr_mb != -1) { in execute_backward_weights_nspc()1210 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local1233 acc_data_t *acc = need_reduction in execute_backward_weights_ncsp()1279 if (need_reduction && dnnl_thr_syncable()) { in execute_backward_weights_ncsp()1315 const int need_reduction = nthr_mb != 1; in execute_backward_weights_ncsp() local[all …]
1760 const bool need_reduction = nthr_NS > 1; in exec_fwd_step_stats() local1766 if (!need_reduction) return; in exec_fwd_step_stats()1783 acc_data_t *r_mean = need_reduction ? rbuf : mean; in exec_fwd_step_stats()1801 c.do_normalise = !need_reduction; in exec_fwd_step_stats()1809 acc_data_t *r_var = need_reduction ? rbuf : var; in exec_fwd_step_stats()1828 c.do_normalise = !need_reduction; in exec_fwd_step_stats()1925 const bool need_reduction = nthr_NS > 1; in exec_bwd_step_diff_ss() local1930 acc_data_t *const r_diff_gamma = need_reduction ? rbuf : diff_gamma; in exec_bwd_step_diff_ss()1932 = need_reduction ? rbuf + nthr_NS * size_C_stat : diff_beta; in exec_bwd_step_diff_ss()1935 if (!need_reduction) return; in exec_bwd_step_diff_ss()
2028 need_reduction = absn < n_for_stirling