[XLA] Fix the indentation for printing fusion computation.

Change: 153502127

1 files changed, 12 insertions, 2 deletions

diff --git a/tensorflow/compiler/xla/service/hlo_computation.cc b/tensorflow/compiler/xla/service/hlo_computation.cc
index 655546d715..526c1c58b3 100644
--- a/tensorflow/compiler/xla/service/hlo_computation.cc
+++ b/tensorflow/compiler/xla/service/hlo_computation.cc
@@ -330,17 +330,27 @@ std::list<HloComputation*> HloComputation::MakeEmbeddedComputationsList()
   return post_order;
 }
 
-string HloComputation::ToString() const {
+string HloComputation::ToString(int nested_level) const {
   std::ostringstream s;
+  for (int i = 0; i < nested_level; i++) {
+    s << "    ";
+  }
   s << name() << " " << ShapeUtil::HumanString(ComputeProgramShape())
     << " { \n";
   for (const HloInstruction* instruction : MakeInstructionPostOrder()) {
+    for (int i = 0; i < nested_level; i++) {
+      s << "    ";
+    }
     s << "  " << instruction->ToString() << "\n";
     if (instruction->opcode() == HloOpcode::kFusion) {
-      s << "    " << instruction->fused_instructions_computation()->ToString()
+      s << instruction->fused_instructions_computation()->ToString(
+               nested_level + 1)
         << "\n";
     }
   }
+  for (int i = 0; i < nested_level; i++) {
+    s << "    ";
+  }
   s << "}";
   return s.str();
 }