Searched refs:softmax_v2 (Results 1 – 4 of 4) sorted by relevance
115 y_tf = nn_ops.softmax_v2(x_tf)116 y_tf_last_dim = nn_ops.softmax_v2(x_tf, 1)125 x_neg_axis = nn_ops.softmax_v2(arr, axis=-2)126 y_pos_axis = nn_ops.softmax_v2(arr, axis=0)127 z_gt_axis = nn_ops.softmax_v2(arr, axis=0)153 y_tf = nn_ops.softmax_v2(x_tf)
3710 def softmax_v2(logits, axis=None, name=None): function
57 activation_map = {nn.softmax_v2: 'softmax'}
305 keras.layers.Dense(y_train.shape[-1], activation=nn.softmax_v2),