diff --git a/source/api_cc/src/common.cc b/source/api_cc/src/common.cc index 203e5f9db1..12309df6a0 100644 --- a/source/api_cc/src/common.cc +++ b/source/api_cc/src/common.cc @@ -996,6 +996,7 @@ template void deepmd::session_get_vector(std::vector&, Session*, const std::string, const std::string); +#endif template void deepmd::select_map(std::vector& out, const std::vector& in, @@ -1025,6 +1026,7 @@ template void deepmd::select_map_inv( const std::vector& idx_map, const int& stride); +#ifdef BUILD_TENSORFLOW template float deepmd::session_get_scalar(Session*, const std::string, const std::string);