Searched refs:softmax_v2 (Results 1 – 5 of 5) sorted by relevance
58 y_tf = nn_ops.softmax_v2(x_tf)74 y_tf = nn_ops.softmax_v2(x_tf)90 y_tf = nn_ops.softmax_v2(x_tf)
1031 @dispatch.dispatch_for_api(nn_ops.softmax_v2)
1022 op=nn_ops.softmax_v2,
112 y_tf = nn_ops.softmax_v2(x_tf)113 y_tf_last_dim = nn_ops.softmax_v2(x_tf, 1)122 x_neg_axis = nn_ops.softmax_v2(arr, axis=-2)123 y_pos_axis = nn_ops.softmax_v2(arr, axis=0)124 z_gt_axis = nn_ops.softmax_v2(arr, axis=0)149 nn_ops.softmax_v2, [x_tf])
3862 def softmax_v2(logits, axis=None, name=None): function3911 softmax.__doc__ = softmax_v2.__doc__