Searched refs:softmax_v2 (Results 1 – 4 of 4) sorted by relevance
111 y_tf = nn_ops.softmax_v2(x_tf)112 y_tf_last_dim = nn_ops.softmax_v2(x_tf, 1)121 x_neg_axis = nn_ops.softmax_v2(arr, axis=-2)122 y_pos_axis = nn_ops.softmax_v2(arr, axis=0)123 z_gt_axis = nn_ops.softmax_v2(arr, axis=0)137 y_tf = nn_ops.softmax_v2(x_tf)
2904 def softmax_v2(logits, axis=None, name=None): function
51 activation_map = {nn.softmax_v2: 'softmax'}
230 keras.layers.Dense(y_train.shape[-1], activation=nn.softmax_v2),