Searched refs:log_softmax_v2 (Results 1 – 2 of 2) sorted by relevance
217 y_tf = nn_ops.log_softmax_v2(x_tf)224 x_neg_axis = nn_ops.log_softmax_v2(arr, axis=-2)225 y_pos_axis = nn_ops.log_softmax_v2(arr, axis=0)226 z_gt_axis = nn_ops.log_softmax_v2(arr, axis=0)240 y_tf = nn_ops.log_softmax_v2(x_tf)
3769 def log_softmax_v2(logits, axis=None, name=None): function