From 1ac0bffaf7d52178b037f4506c3d60d5d49241f4 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 18 Jul 2017 13:11:06 +0800
Subject: [PATCH] Override Net::DebugString to invoke inner ops_

Net::DebugString() print what it is and invokes inner operators
DebugString() with a prefix '\t'.
---
 paddle/framework/net.cc     | 9 +++++++++
 paddle/framework/net.h      | 2 ++
 paddle/framework/operator.h | 2 +-
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/paddle/framework/net.cc b/paddle/framework/net.cc
index 7311cda9a9..b9cd732d40 100644
--- a/paddle/framework/net.cc
+++ b/paddle/framework/net.cc
@@ -55,5 +55,14 @@ void PlainNet::CompleteAddOp() {
   add_op_done_ = true;
 }
 
+std::string PlainNet::DebugString() const {
+  std::ostringstream os;
+  os << this->type_ << ":" << std::endl;
+  for (auto& op : ops_) {
+    os << "\t" << op->DebugString() << std::endl;
+  }
+  return os.str();
+}
+
 }  // namespace framework
 }  // namespace paddle
diff --git a/paddle/framework/net.h b/paddle/framework/net.h
index 19a1620e29..33bb30ea07 100644
--- a/paddle/framework/net.h
+++ b/paddle/framework/net.h
@@ -88,6 +88,8 @@ class PlainNet : public Net {
 
   void CompleteAddOp() override;
 
+  std::string DebugString() const override;
+
   std::vector<OperatorPtr> ops_;
 
  private:
diff --git a/paddle/framework/operator.h b/paddle/framework/operator.h
index f7ed6e9f3d..b62cac6d27 100644
--- a/paddle/framework/operator.h
+++ b/paddle/framework/operator.h
@@ -48,7 +48,7 @@ class OperatorBase {
     return boost::get<T>(attrs_.at(name));
   }
 
-  std::string DebugString() const;
+  virtual std::string DebugString() const;
 
   /// Init will be called after CreateOperator, you can put some initialization
   /// logic here.
-- 
GitLab