Searched refs:sum_exp_logits (Results 1 – 1 of 1) sorted by relevance
61 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentLossGenerator() argument65 sum_exp_logits_(sum_exp_logits), in SparseXentLossGenerator()101 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentGradGenerator() argument105 sum_exp_logits_(sum_exp_logits), in SparseXentGradGenerator()