[XLA] [NFC] Remove prints from test

PiperOrigin-RevId: 397195421 Change-Id: I0daa898f8e9fd2848eccfd43bb7719ba91bce207
CynthiaCYX · Sep 16, 2021 · 7163b65 · 7163b65
1 parent 64baf86
commit 7163b65
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tensorflow/compiler/xla/service/layout_assignment_test.cc b/tensorflow/compiler/xla/service/layout_assignment_test.cc
@@ -1511,7 +1511,6 @@ ENTRY main {
 
   TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m,
                           ParseAndReturnVerifiedModule(module_str));
-  std::cerr << m->ToString();
   ComputationLayout computation_layout(
       m->entry_computation()->ComputeProgramShape());
   *computation_layout.mutable_parameter_layout(0) =
@@ -1520,14 +1519,12 @@ ENTRY main {
       ShapeLayout(ShapeUtil::MakeShapeWithLayout(F32, {93184, 4}, {0, 1}));
   *computation_layout.mutable_result_layout() = ShapeLayout(
       ShapeUtil::MakeShapeWithLayout(F32, {2, 512, 364}, {0, 1, 2}));
-  std::cerr << computation_layout.ToString();
   ChannelLayoutConstraints channel_constraints;
   LayoutAssignment layout_assignment(
       &computation_layout,
       /*channel_constraints=*/&channel_constraints,
       /* reverse_computation_order = */ true);
   EXPECT_IS_OK(layout_assignment.Run(m.get()).status());
-  std::cerr << m->ToString();
   const HloInstruction* call_1 = FindInstruction(m.get(), "reshape.8494");
   ExpectLayoutIs(call_1->shape(), {0, 1, 2});
   const HloInstruction* on_true = FindInstruction(m.get(), "reshape.8493");