ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
This is the complete list of members for KEDF_ML, including all inherited members.
cal_tool | KEDF_ML | |
chi_p | KEDF_ML | |
chi_pnl | KEDF_ML | |
chi_q | KEDF_ML | |
chi_qnl | KEDF_ML | |
chi_xi | KEDF_ML | |
cTF | KEDF_ML | |
descriptor2index | KEDF_ML | |
descriptor2kernel | KEDF_ML | |
descriptor_type | KEDF_ML | |
device | KEDF_ML | |
device_CPU | KEDF_ML | |
device_type | KEDF_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 | |
dV | KEDF_ML | |
enhancement_cpu_ptr | KEDF_ML | |
feg3_correct | KEDF_ML | |
feg_net_F | KEDF_ML | |
gamma | KEDF_ML | |
gammanl | KEDF_ML | |
gene_data_label | KEDF_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_ptr | KEDF_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_ML | inline |
kernel_index | KEDF_ML | |
loadVector(std::string filename, std::vector< double > &data) | KEDF_ML | |
localTest(const double *const *prho, ModulePW::PW_Basis *pw_rho) | KEDF_ML | |
ml_energy | KEDF_ML | |
ml_gamma | KEDF_ML | |
ml_gammanl | KEDF_ML | |
ml_p | KEDF_ML | |
ml_pnl | KEDF_ML | |
ml_potential(const double *const *prho, ModulePW::PW_Basis *pw_rho, ModuleBase::matrix &rpotential) | KEDF_ML | |
ml_q | KEDF_ML | |
ml_qnl | KEDF_ML | |
ml_tanh_pnl | KEDF_ML | |
ml_tanh_qnl | KEDF_ML | |
ml_tanhp | KEDF_ML | |
ml_tanhp_nl | KEDF_ML | |
ml_tanhq | KEDF_ML | |
ml_tanhq_nl | KEDF_ML | |
ml_tanhxi | KEDF_ML | |
ml_tanhxi_nl | KEDF_ML | |
ml_xi | KEDF_ML | |
nablaRho | KEDF_ML | |
ninput | KEDF_ML | |
nkernel | KEDF_ML | |
nn | KEDF_ML | |
NN_forward(const double *const *prho, ModulePW::PW_Basis *pw_rho, bool cal_grad) | KEDF_ML | |
nx | KEDF_ML | |
nx_tot | KEDF_ML | |
p | KEDF_ML | |
pnl | KEDF_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 | |
pqcoef | KEDF_ML | |
q | KEDF_ML | |
qnl | KEDF_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_pnl | KEDF_ML | |
tanh_qnl | KEDF_ML | |
tanhp | KEDF_ML | |
tanhp_nl | KEDF_ML | |
tanhq | KEDF_ML | |
tanhq_nl | KEDF_ML | |
tanhxi | KEDF_ML | |
tanhxi_nl | KEDF_ML | |
updateInput(const double *const *prho, ModulePW::PW_Basis *pw_rho) | KEDF_ML | |
xi | KEDF_ML | |
~KEDF_ML() | KEDF_ML | inline |