Searched defs:SubtractAndExpFunctor (Results 1 – 1 of 1) sorted by relevance
97 struct SubtractAndExpFunctor { struct98 __host__ __device__ SubtractAndExpFunctor(const T* logits, in SubtractAndExpFunctor() argument103 __host__ __device__ U operator()(const int gid) const { in operator ()()110 const T* logits_;111 const T* max_logits_;112 const int num_cols_;