File prod_env_mat.h

↰ Parent directory (source/lib/include)

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

• Program Listing for File prod_env_mat.h

Includes

• device.h (File device.h)

• neighbor_list.h (File neighbor_list.h)

• vector

Namespaces

• Namespace deepmd

Functions

• Function deepmd::env_mat_nbor_update

• 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 int, const float, const float, const std::vector<int>, const int *)

• 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 *)

• Template Function deepmd::prod_env_mat_a_gpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, int *, unsigned long long *, const int, const FPTYPE *, const FPTYPE *, const int, const int, const int, const float, const float, const std::vector<int>, const int *)

• Template Function deepmd::prod_env_mat_a_gpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, int *, unsigned long long *, const int, const FPTYPE *, const FPTYPE *, const int, const int, const float, const float, const std::vector<int>, const int *)

• Template Function deepmd::prod_env_mat_r_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>)

• Template Function deepmd::prod_env_mat_r_cpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, const int, const FPTYPE *, const FPTYPE *, const int, const int, const int, const float, const float, const std::vector<int>)

• Template Function deepmd::prod_env_mat_r_gpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, int *, unsigned long long *, const int, const FPTYPE *, const FPTYPE *, const int, const int, const float, const float, const std::vector<int>)

• Template Function deepmd::prod_env_mat_r_gpu(FPTYPE *, FPTYPE *, FPTYPE *, int *, const FPTYPE *, const int *, const InputNlist&, int *, unsigned long long *, const int, const FPTYPE *, const FPTYPE *, const int, const int, const int, const float, const float, const std::vector<int>)