Searched refs:log_sum_exp (Results 1 – 2 of 2) sorted by relevance
146 auto log_sum_exp = xla::Log(sum_exp); in CrossEntropyWithLogits() local152 auto sub = xla::Sub(shifted_logits, log_sum_exp, {kBatchDim}); in CrossEntropyWithLogits()
4147 const float log_sum_exp = std::log(sum_exp); in LogSoftmax() local4152 (input_scale * (max_val + zero_point_offset) + log_sum_exp) / in LogSoftmax()