-
Avelina9X authored
* Added softmax_dtype argument to coerce log_softmax computations * move softmax_dtype --------- Co-authored-by:Baber <baber@hey.com>
e4a7b69f
* Added softmax_dtype argument to coerce log_softmax computations
* move softmax_dtype
---------
Co-authored-by: Baber <baber@hey.com>