Searched defs:GetSumExpFloat (Results 1 – 1 of 1) sorted by relevance
100 struct GetSumExpFloat { struct101 __device__ __forceinline__ GetSumExpFloat(AccT v) : max_k(v) {} in GetSumExpFloat() function103 …__ __forceinline__ AccT operator()(AccT sum, T v) const { return sum + std::exp((AccT)v - max_k); } in operator ()()105 const AccT max_k;