Searched refs:sum_exp_logits_ (Results 1 – 1 of 1) sorted by relevance
65 sum_exp_logits_(sum_exp_logits), in SparseXentLossGenerator()78 ? (Eigen::numext::log(sum_exp_logits_(batch)) - logits_(coords)) in operator()84 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits_; variable105 sum_exp_logits_(sum_exp_logits), in SparseXentGradGenerator()118 return exp_logits_(coords) / sum_exp_logits_(batch) - subtract; in operator()123 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits_; variable