Searched defs:sum_exp (Results 1 – 4 of 4) sorted by relevance
132 float sum_exp = 0.0; in TestF16() local196 double sum_exp = 0.0; in TestF32() local259 float sum_exp = 0.0f; in TestQU8() local
72 auto sum_exp = XlaHelpers::ConvertElementType(reduce, type); in CrossEntropyWithLogits() local
358 T sum_exp = 0; in CalculateRowSoftmax() local
3465 float sum_exp = 0.0f; in Softmax() local3637 int32 sum_exp = 0; in SoftmaxInt8LUT() local3820 float sum_exp = 0.0f; in LogSoftmax() local