Searched refs:sum_exp (Results 1 – 1 of 1) sorted by relevance
118 auto sum_exp = XlaHelpers::ConvertElementType(reduce, type); in CrossEntropyWithLogits() local121 auto log_sum_exp = xla::Log(sum_exp); in CrossEntropyWithLogits()138 xla::Sub(xla::Div(exp_shifted_logits, sum_exp, {kBatchDim}), labels); in CrossEntropyWithLogits()