ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
This is the complete list of members for NN_OFImpl, including all inherited members.
F | NN_OFImpl | |
fc1 | NN_OFImpl | |
fc2 | NN_OFImpl | |
fc3 | NN_OFImpl | |
fc4 | NN_OFImpl | |
forward(torch::Tensor inpt) | NN_OFImpl | |
input_vali | NN_OFImpl | |
inputs | NN_OFImpl | |
nfc | NN_OFImpl | |
ninpt | NN_OFImpl | |
nlayer | NN_OFImpl | |
NN_OFImpl(int nrxx, int nrxx_vali, int ninpt, int nnode, int nlayer, torch::Device device) | NN_OFImpl | |
nnode | NN_OFImpl | |
nrxx | NN_OFImpl | |
nrxx_vali | NN_OFImpl | |
set_data(T *data, const std::vector< std::string > &descriptor_type, const std::vector< int > &kernel_index, torch::Tensor &nn_input) | NN_OFImpl | inline |
~NN_OFImpl() | NN_OFImpl | inline |