Home
last modified time | relevance | path

Searched refs:log_softmax_v2 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_test.py203 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])
H A Dnn_ops.py3948 def log_softmax_v2(logits, axis=None, name=None): function
4145 log_probs = log_softmax_v2(precise_logits)
4255 log_probs = log_softmax_v2(logits)