Searched defs:softmax_lse_d (Results 1 – 1 of 1) sorted by relevance
68 void *softmax_lse_d, in set_params_fprop()167 void set_params_dgrad(Flash_bwd_params ¶ms, in set_params_dgrad()