Searched refs:log_softmax_v2 (Results 1 – 2 of 2) sorted by relevance
203 y_tf = nn_ops.log_softmax_v2(x_tf)210 x_neg_axis = nn_ops.log_softmax_v2(arr, axis=-2)211 y_pos_axis = nn_ops.log_softmax_v2(arr, axis=0)212 z_gt_axis = nn_ops.log_softmax_v2(arr, axis=0)226 nn_ops.log_softmax_v2, [x_tf])
3948 def log_softmax_v2(logits, axis=None, name=None): function4145 log_probs = log_softmax_v2(precise_logits)4255 log_probs = log_softmax_v2(logits)