Template Function deepmd::tabulate_fusion_se_a_grad_gpu

• Defined in File tabulate.h

Function Documentation

template<typename FPTYPE>
void deepmd::tabulate_fusion_se_a_grad_gpu(FPTYPE *dy_dem_x, FPTYPE *dy_dem, FPTYPE *dy_dtwo, const FPTYPE *table, const FPTYPE *table_info, const FPTYPE *em_x, const FPTYPE *em, const FPTYPE *two_embed, const FPTYPE *dy, const int nloc, const int nnei, const int last_layer_size, const bool is_sorted = true)