File gelu.h

↰ Parent directory (source/lib/include)

Definition (source/lib/include/gelu.h)

• Program Listing for File gelu.h

Includes

• device.h (File device.h)

Namespaces

• Namespace deepmd

Functions

• Template Function deepmd::gelu_cpu

• Template Function deepmd::gelu_gpu

• Template Function deepmd::gelu_grad_cpu

• Template Function deepmd::gelu_grad_gpu

• Template Function deepmd::gelu_grad_grad_cpu

• Template Function deepmd::gelu_grad_grad_gpu