split fvar version of trace_quad_form and trace_gen_quad_form from math version
fixed log_softmax using statements