From 6415b0771b924f0fc8abb49660ffca20e0d36a44 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Fri, 24 Feb 2017 13:02:09 +0800
Subject: [PATCH] Add getNames

---
 paddle/api/Evaluator.cpp | 6 ++++++
 paddle/api/PaddleAPI.h   | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/paddle/api/Evaluator.cpp b/paddle/api/Evaluator.cpp
index c30e0987639..f9656db19a0 100644
--- 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;
+}
diff --git a/paddle/api/PaddleAPI.h b/paddle/api/PaddleAPI.h
index f5af8b0035b..f5dcfcf94c6 100644
--- 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;
 
-- 
GitLab