Template Function deepmd::prod_force_a_gpu

• Defined in File prod_force.h

Function Documentation

template<typename FPTYPE>
void deepmd::prod_force_a_gpu(FPTYPE *force, const FPTYPE *net_deriv, const FPTYPE *in_deriv, const int *nlist, const int nloc, const int nall, const int nnei, const int nframes)