Searched refs:ilabel_log_probs (Results 1 – 1 of 1) sorted by relevance
550 def _ilabel_to_state(labels, num_labels, ilabel_log_probs): argument554 blank = ilabel_log_probs[:, :, :1]558 ilabel_log_probs = array_ops.expand_dims(ilabel_log_probs, axis=2)559 state_log_probs = math_ops.reduce_sum(ilabel_log_probs * one_hot, axis=3)656 ilabel_log_probs = nn_ops.log_softmax(logits)657 state_log_probs = _ilabel_to_state(labels, num_labels, ilabel_log_probs)674 grad = math_ops.exp(ilabel_log_probs) - math_ops.exp(olabel_log_probs)