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

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

FNN_OFImpl
fc1NN_OFImpl
fc2NN_OFImpl
fc3NN_OFImpl
fc4NN_OFImpl
forward(torch::Tensor inpt)NN_OFImpl
input_valiNN_OFImpl
inputsNN_OFImpl
nfcNN_OFImpl
ninptNN_OFImpl
nlayerNN_OFImpl
NN_OFImpl(int nrxx, int nrxx_vali, int ninpt, int nnode, int nlayer, torch::Device device)NN_OFImpl
nnodeNN_OFImpl
nrxxNN_OFImpl
nrxx_valiNN_OFImpl
set_data(T *data, const std::vector< std::string > &descriptor_type, const std::vector< int > &kernel_index, torch::Tensor &nn_input)NN_OFImplinline
~NN_OFImpl()NN_OFImplinline