Searched defs:LogSoftMaxForwardEpilogue (Results 1 – 1 of 1) sorted by relevance
40 struct LogSoftMaxForwardEpilogue { struct41 __device__ __forceinline__ LogSoftMaxForwardEpilogue(AccumT max_input, AccumT sum) in LogSoftMaxForwardEpilogue() argument44 __device__ __forceinline__ OutT operator()(T input) const { in operator ()()48 const AccumT max_input;49 const AccumT logsum;