Template Function deepmd::tabulate_fusion_se_a_grad_grad_gpu
Defined in File tabulate.h
Function Documentation
- template<typename FPTYPE>
void deepmd::tabulate_fusion_se_a_grad_grad_gpu(FPTYPE *dz_dy, const FPTYPE *table, const FPTYPE *table_info, const FPTYPE *em_x, const FPTYPE *em, const FPTYPE *two_embed, const FPTYPE *dz_dy_dem_x, const FPTYPE *dz_dy_dem, const FPTYPE *dz_dy_dtwo, const int nloc, const int nnei, const int last_layer_size, const bool is_sorted = true)