Searched refs:log_softmax_v2 (Results 1 – 2 of 2) sorted by relevance
201 y_tf = nn_ops.log_softmax_v2(x_tf)208 x_neg_axis = nn_ops.log_softmax_v2(arr, axis=-2)209 y_pos_axis = nn_ops.log_softmax_v2(arr, axis=0)210 z_gt_axis = nn_ops.log_softmax_v2(arr, axis=0)224 y_tf = nn_ops.log_softmax_v2(x_tf)
2961 def log_softmax_v2(logits, axis=None, name=None): function