Template Function deepmd::prod_env_mat_a_cpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, const int, const FPTYPE *, const FPTYPE *, const int, const int, const float, const float, const std::vector<int>, const int *)#
Defined in File prod_env_mat.h
Function Documentation#
- template<typename FPTYPE>
inline void deepmd::prod_env_mat_a_cpu(FPTYPE *em, FPTYPE *em_deriv, FPTYPE *rij, int *nlist, const FPTYPE *coord, const int *type, const InputNlist &inlist, const int max_nbor_size, const FPTYPE *avg, const FPTYPE *std, const int nloc, const int nall, const float rcut, const float rcut_smth, const std::vector<int> sec, const int *f_type = NULL)#