Searched refs:dwei_ptr (Results 1 – 4 of 4) sorted by relevance
913 diff_wei_data_t *dwei_ptr in bf16_bwd_weights_reduction_par_nspc() local916 (bfloat16_t *)(dwei_ptr), wei_reduced, ws_ptr, jcp.oc); in bf16_bwd_weights_reduction_par_nspc()
904 diff_wei_data_t *dwei_ptr in bf16_bwd_weights_reduction_par_nspc() local907 (bfloat16_t *)(dwei_ptr), wei_reduced, ws_ptr, jcp.oc); in bf16_bwd_weights_reduction_par_nspc()
2177 float *__restrict dwei_ptr in bwd_weights_reduction_par_nspc() local2183 dwei_ptr[oc] = ws_ptr[oc]; in bwd_weights_reduction_par_nspc()2188 dwei_ptr[oc] += ws_ptr[oc]; in bwd_weights_reduction_par_nspc()
2163 float *__restrict dwei_ptr in bwd_weights_reduction_par_nspc() local2169 dwei_ptr[oc] = ws_ptr[oc]; in bwd_weights_reduction_par_nspc()2174 dwei_ptr[oc] += ws_ptr[oc]; in bwd_weights_reduction_par_nspc()