Searched defs:p_dropout (Results 1 – 2 of 2) sorted by relevance
69 float p_dropout, in set_params_fprop()167 void set_params_dgrad(Flash_bwd_params ¶ms, in set_params_dgrad()299 const int head_size_rounded, const float p_dropout, in set_params_splitkv()355 const float p_dropout, in mha_fwd()554 const float p_dropout, in mha_varlen_fwd()817 const float p_dropout, // probability to drop in mha_bwd()1032 const float p_dropout, // probability to drop in mha_varlen_bwd()
109 float p_dropout; member