Template Function deepmd::tabulate_fusion_se_a_gpu

Function Documentation

template<typename FPTYPE>
void deepmd::tabulate_fusion_se_a_gpu(FPTYPE *out, const FPTYPE *table, const FPTYPE *table_info, const FPTYPE *em_x, const FPTYPE *em, const FPTYPE *two_embed, const int nloc, const int nnei, const int last_layer_size, const bool is_sorted = true)