Searched refs:wei_reduction (Results 1 – 10 of 10) sorted by relevance
684 auto wei_reduction = scratchpad.template get<float>(key_conv_wei_reduction); in execute_backward_weights() local698 auto bia_reduction = wei_reduction + n_wei_buffers * wei_size; in execute_backward_weights()771 diff_wei = wei_reduction + (ithr_mb)*wei_size; in execute_backward_weights()775 : (float *)wei_reduction + (ithr_mb - 1) * wei_size; in execute_backward_weights()1018 ? wei_reduction + off in execute_backward_weights()1022 float *wei_to_reduce = wei_reduction in execute_backward_weights()1086 (const float *)(wei_reduction + off), acc_size); in execute_backward_weights()
974 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_nspc() local1043 acc_data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()1132 acc_data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()1180 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_ncsp() local1237 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()1333 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()
651 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local831 : wei_reduction + (ithr_mb - 1) * wei_size; in execute_backward_weights()989 data_t *s = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()1114 = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()
687 auto wei_reduction = scratchpad.template get<float>(key_conv_wei_reduction); in execute_backward_weights() local701 auto bia_reduction = wei_reduction + n_wei_buffers * wei_size; in execute_backward_weights()774 diff_wei = wei_reduction + (ithr_mb)*wei_size; in execute_backward_weights()778 : (float *)wei_reduction + (ithr_mb - 1) * wei_size; in execute_backward_weights()1021 ? wei_reduction + off in execute_backward_weights()1025 float *wei_to_reduce = wei_reduction in execute_backward_weights()1089 (const float *)(wei_reduction + off), acc_size); in execute_backward_weights()
965 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_nspc() local1034 acc_data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()1123 acc_data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()1171 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_ncsp() local1228 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()1324 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()
653 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local833 : wei_reduction + (ithr_mb - 1) * wei_size; in execute_backward_weights()991 data_t *s = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()1116 = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()
600 auto wei_reduction in execute_backward_weights_nspc() local648 data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()727 data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()769 auto wei_reduction in execute_backward_weights_ncsp() local812 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()894 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()
597 auto wei_reduction in execute_backward_weights_nspc() local645 data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()724 data_t *weights_reduce_base = wei_reduction in execute_backward_weights_nspc()766 auto wei_reduction in execute_backward_weights_ncsp() local809 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()891 = wei_reduction + ithr_g * nthr_mb * weights_g_size; in execute_backward_weights_ncsp()
486 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local593 : wei_reduction + (ithr_mb - 1) * wei_size; in execute_backward_weights()734 data_t *s = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()859 = wei_reduction + (thr_mb - 1) * wei_size + off; in execute_backward_weights()