ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
KEDF_ML Member List

This is the complete list of members for KEDF_ML, including all inherited members.

cal_toolKEDF_ML
chi_pKEDF_ML
chi_pnlKEDF_ML
chi_qKEDF_ML
chi_qnlKEDF_ML
chi_xiKEDF_ML
cTFKEDF_ML
descriptor2indexKEDF_ML
descriptor2kernelKEDF_ML
descriptor_typeKEDF_ML
deviceKEDF_ML
device_CPUKEDF_ML
device_typeKEDF_ML
dumpMatrix(std::string filename, const ModuleBase::matrix &data)KEDF_ML
dumpTensor(std::string filename, const torch::Tensor &data)KEDF_ML
dumpVector(std::string filename, const std::vector< double > &data)KEDF_ML
dVKEDF_ML
enhancement_cpu_ptrKEDF_ML
feg3_correctKEDF_ML
feg_net_FKEDF_ML
gammaKEDF_ML
gammanlKEDF_ML
gene_data_labelKEDF_ML
generateTrainData(const double *const *prho, ModulePW::PW_Basis *pw_rho, const double *veff)KEDF_ML
get_data(std::string parameter, const int ikernel)KEDF_ML
get_energy(const double *const *prho, ModulePW::PW_Basis *pw_rho)KEDF_ML
get_potential_(const double *const *prho, ModulePW::PW_Basis *pw_rho, ModuleBase::matrix &rpotential)KEDF_ML
gradient_cpu_ptrKEDF_ML
init_data(const int &nkernel, const bool &of_ml_gamma, const bool &of_ml_p, const bool &of_ml_q, const bool &of_ml_tanhp, const bool &of_ml_tanhq, const std::vector< int > &of_ml_gammanl_, const std::vector< int > &of_ml_pnl, const std::vector< int > &of_ml_qnl, const std::vector< int > &of_ml_xi, const std::vector< int > &of_ml_tanhxi, const std::vector< int > &of_ml_tanhxi_nl, const std::vector< int > &of_ml_tanh_pnl, const std::vector< int > &of_ml_tanh_qnl, const std::vector< int > &of_ml_tanhp_nl, const std::vector< int > &of_ml_tanhq_nl)KEDF_ML
KEDF_ML()KEDF_MLinline
kernel_indexKEDF_ML
loadVector(std::string filename, std::vector< double > &data)KEDF_ML
localTest(const double *const *prho, ModulePW::PW_Basis *pw_rho)KEDF_ML
ml_energyKEDF_ML
ml_gammaKEDF_ML
ml_gammanlKEDF_ML
ml_pKEDF_ML
ml_pnlKEDF_ML
ml_potential(const double *const *prho, ModulePW::PW_Basis *pw_rho, ModuleBase::matrix &rpotential)KEDF_ML
ml_qKEDF_ML
ml_qnlKEDF_ML
ml_tanh_pnlKEDF_ML
ml_tanh_qnlKEDF_ML
ml_tanhpKEDF_ML
ml_tanhp_nlKEDF_ML
ml_tanhqKEDF_ML
ml_tanhq_nlKEDF_ML
ml_tanhxiKEDF_ML
ml_tanhxi_nlKEDF_ML
ml_xiKEDF_ML
nablaRhoKEDF_ML
ninputKEDF_ML
nkernelKEDF_ML
nnKEDF_ML
NN_forward(const double *const *prho, ModulePW::PW_Basis *pw_rho, bool cal_grad)KEDF_ML
nxKEDF_ML
nx_totKEDF_ML
pKEDF_ML
pnlKEDF_ML
potGammanlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rGammanlTerm)KEDF_ML
potGammaTerm(int ir)KEDF_ML
potPPnlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rPPnlTerm)KEDF_ML
potPTerm1(int ir)KEDF_ML
potQQnlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rQQnlTerm)KEDF_ML
potQTerm1(int ir)KEDF_ML
potTanhpTanh_pnlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhpTanh_pnlTerm)KEDF_ML
potTanhpTanhp_nlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhpTanhp_nlTerm)KEDF_ML
potTanhpTerm1(int ir)KEDF_ML
potTanhqTanh_qnlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhqTanh_qnlTerm)KEDF_ML
potTanhqTanhq_nlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhqTanhq_nlTerm)KEDF_ML
potTanhqTerm1(int ir)KEDF_ML
potTanhxi_nlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhxi_nlTerm)KEDF_ML
potTanhxinlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rTanhxinlTerm)KEDF_ML
potTanhxiTerm1(int ir)KEDF_ML
potXinlTerm(const double *const *prho, ModulePW::PW_Basis *pw_rho, std::vector< double > &rXinlTerm)KEDF_ML
potXiTerm1(int ir)KEDF_ML
pqcoefKEDF_ML
qKEDF_ML
qnlKEDF_ML
set_device(std::string device_inpt)KEDF_ML
set_para(const int nx, const double dV, const double nelec, const double tf_weight, const double vw_weight, const double chi_p, const double chi_q, const std::vector< double > &chi_xi, const std::vector< double > &chi_pnl, const std::vector< double > &chi_qnl, const int &nkernel, const std::vector< int > &kernel_type, const std::vector< double > &kernel_scaling, const std::vector< double > &yukawa_alpha, const std::vector< std::string > &kernel_file, const bool &of_ml_gamma, const bool &of_ml_p, const bool &of_ml_q, const bool &of_ml_tanhp, const bool &of_ml_tanhq, const std::vector< int > &of_ml_gammanl, const std::vector< int > &of_ml_pnl, const std::vector< int > &of_ml_qnl, const std::vector< int > &of_ml_xi, const std::vector< int > &of_ml_tanhxi, const std::vector< int > &of_ml_tanhxi_nl, const std::vector< int > &of_ml_tanh_pnl, const std::vector< int > &of_ml_tanh_qnl, const std::vector< int > &of_ml_tanhp_nl, const std::vector< int > &of_ml_tanhq_nl, const std::string device_inpt, ModulePW::PW_Basis *pw_rho)KEDF_ML
tanh_pnlKEDF_ML
tanh_qnlKEDF_ML
tanhpKEDF_ML
tanhp_nlKEDF_ML
tanhqKEDF_ML
tanhq_nlKEDF_ML
tanhxiKEDF_ML
tanhxi_nlKEDF_ML
updateInput(const double *const *prho, ModulePW::PW_Basis *pw_rho)KEDF_ML
xiKEDF_ML
~KEDF_ML()KEDF_MLinline