Searched defs:LogSoftMaxBackwardEpilogue (Results 1 – 1 of 1) sorted by relevance
53 struct LogSoftMaxBackwardEpilogue { struct54 __device__ __forceinline__ LogSoftMaxBackwardEpilogue(AccumT sum) in LogSoftMaxBackwardEpilogue() argument57 __device__ __forceinline__ T operator()(OutT gradOutput, OutT output) const { in operator ()()61 const AccumT sum;