From bdf5a88b8f0a8b06b5fa95c1d5540651db889406 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 21 Dec 2017 11:11:22 -0800 Subject: [PATCH] [XLA] Make dumped_computation_to_text print the short parsable hlo text. PiperOrigin-RevId: 179835513 --- tensorflow/compiler/xla/tools/dumped_computation_to_text.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc b/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc index 78d8fb1f43..24417a0cb8 100644 --- a/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc +++ b/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc @@ -69,7 +69,7 @@ void RealMain(tensorflow::gtl::ArraySlice args, bool compile) { fprintf(stdout, "HLO compiled for %s backend:\n%s\n", local_service->backend().platform()->Name().c_str(), - module.ToString().c_str()); + module.ToString(HloPrintOptions::ShortParsable()).c_str()); } else { const ComputationTracker& tracker = local_service->computation_tracker(); UserComputation* user_computation = @@ -80,7 +80,8 @@ void RealMain(tensorflow::gtl::ArraySlice args, bool compile) { tracker.BuildHloModule(versioned_handle, HloModuleConfig()) .ConsumeValueOrDie(); - fprintf(stdout, "%s\n", module->ToString().c_str()); + fprintf(stdout, "%s\n", + module->ToString(HloPrintOptions::ShortParsable()).c_str()); } } } -- 2.34.1