Template Function deepmd::gelu_grad_gpu

Template Function deepmd::gelu_grad_gpu#

Function Documentation#

template<typename FPTYPE>
void deepmd::gelu_grad_gpu(FPTYPE *out, const FPTYPE *xx, const FPTYPE *dy, const int_64 size)#