Searched refs:ilabel_log_probs (Results 1 – 1 of 1) sorted by relevance
455 def _ilabel_to_state(labels, num_labels, ilabel_log_probs): argument459 blank = ilabel_log_probs[:, :, :1]463 ilabel_log_probs = array_ops.expand_dims(ilabel_log_probs, axis=2)464 state_log_probs = math_ops.reduce_sum(ilabel_log_probs * one_hot, axis=3)553 ilabel_log_probs = nn_ops.log_softmax(logits)554 state_log_probs = _ilabel_to_state(labels, num_labels, ilabel_log_probs)571 grad = math_ops.exp(ilabel_log_probs) - math_ops.exp(olabel_log_probs)