Searched refs:log_sum_exp (Results 1 – 2 of 2) sorted by relevance
121 auto log_sum_exp = xla::Log(sum_exp); in CrossEntropyWithLogits() local127 auto sub = xla::Sub(shifted_logits, log_sum_exp, {kBatchDim}); in CrossEntropyWithLogits()
299 log_sum_exp = math_ops.reduce_logsumexp(concat_log_probs, [0])300 return log_sum_exp