Searched refs:check_softmax_args (Results 1 – 3 of 3) sorted by relevance
26 bool check_softmax_args(
71 bool check_softmax_args( in check_softmax_args() function
32 check_softmax_args(in, dim, half_to_float, out), in softmax_out()