Home
last modified time | relevance | path

Searched refs:SubtractAndExpFunctor (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsoftmax_op_gpu.cu.cc97 struct SubtractAndExpFunctor { struct
98 __host__ __device__ SubtractAndExpFunctor(const T* logits, in SubtractAndExpFunctor() argument
169 SubtractAndExpFunctor<T, acc_type>, in Compute()
175 SubtractAndExpFunctor<T, acc_type>( in Compute()