Template Function deepmd::gelu_grad_grad_gpu

Defined in File gelu.h

Function Documentation

template<typename FPTYPE> void deepmd::gelu_grad_grad_gpu(FPTYPE *out, const FPTYPE *xx, const FPTYPE *dy, const FPTYPE *dy_2, const int_64 size)

Read the Docs v: master

Versions: master; latest; stable; v2.2.7; v2.2.6; v2.2.5; v2.2.4; v2.2.3; v2.2.2; v2.2.1; v2.2.0; v2.2.0.b0; v2.1.5; v2.1.4; v2.1.3; v2.1.2; v2.1.1; v2.1.0; v2.0.3; v2.0.2; v2.0.1; v2.0.0; v2.0.0.b4; v2.0.0.b3; v2.0.0.b2; v2.0.0.b1; v2.0.0.b0; v2.0.0.a1; v2.0.0.a0; v1.3.3; v1.3.2; v1.3.1; v1.3.0; r1.3; r1.2; devel-20220601; devel-20220531

Downloads: pdf; html; epub

On Read the Docs: Project Home; Builds