Template Function deepmd::gelu_grad_gpu_cuda

Defined in File gelu.h

Function Documentation

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

Read the Docs v: v2.1.5

Versions: master; latest; stable; 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

Downloads

On Read the Docs: Project Home; Builds