Searched defs:return_softmax (Results 1 – 2 of 2) sorted by relevance
70 const bool return_softmax = params.p_ptr != nullptr; in run_flash_fwd() local
360 const bool return_softmax, in mha_fwd()560 const bool return_softmax, in mha_varlen_fwd()