Add getNames

9 years ago · 6415b0771b
parent c45195746b
commit 6415b0771b
2 changed files with 8 additions and 0 deletions
--- a/paddle/api/Evaluator.cpp
+++ b/paddle/api/Evaluator.cpp
@ -27,3 +27,9 @@ std::string Evaluator::toString() {
  m->rawPtr->printStats(sout);
  return sout.str();
 }
+
+std::vector<std::string> Evaluator::getNames() const {
+  std::vector<std::string> retv;
+  m->rawPtr->getNames(&retv);
+  return retv;
+}
--- a/paddle/api/PaddleAPI.h
+++ b/paddle/api/PaddleAPI.h
@ -900,6 +900,8 @@ public:
   */
  std::string toString();

+  std::vector<std::string> getNames() const;
+
 private:
  EvaluatorPrivate* m;