Searched defs:SubtractAndExpFunctor (Results 1 – 1 of 1) sorted by relevance
163 struct SubtractAndExpFunctor { struct164 __host__ __device__ SubtractAndExpFunctor(const T* __restrict__ logits, in SubtractAndExpFunctor() function169 __host__ __device__ U operator()(const int gid) const { in operator ()()176 const T* logits_;177 const T* max_logits_;178 const int num_cols_;