Program Listing for File soft_min_switch_force_grad.h
↰ Return to documentation for file (source/lib/include/soft_min_switch_force_grad.h
)
// SPDX-License-Identifier: LGPL-3.0-or-later
#pragma once
namespace deepmd {
template <typename FPTYPE>
void soft_min_switch_force_grad_cpu(FPTYPE* grad_net,
const FPTYPE* grad,
const FPTYPE* sw_deriv,
const int* nlist,
const int nloc,
const int nnei);
}