experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Remove redundant type info on constants in HLO graph dump.

Browse Source 
Instead of printing constants as e.g. "f32[] 1 (f32[])", print them simply as
"f32[] 1".

PiperOrigin-RevId: 235632525
This commit is contained in:
Justin Lebar 2019-02-25 17:26:05 -08:00 committed by TensorFlower Gardener
parent 9c95751e87
commit 70650633f2
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/service/hlo_graph_dumper.cc
View File

@ -829,8 +829,7 @@ string HloDotDumper::GetInstructionNodeInlinedOperands(
// collected from profiling tools. Those constants may not have a valid // collected from profiling tools. Those constants may not have a valid
// literal. // literal.
if (elem_count.has_value() && *elem_count <= 8 && constant->HasLiteral()) { if (elem_count.has_value() && *elem_count <= 8 && constant->HasLiteral()) {
return StrFormat("%s (%s)", constant->literal().ToString(), return constant->literal().ToString();
ShapeUtil::HumanString(constant->shape()));
} }
// Otherwise, print e.g. "%constant.42 (s32[100])". // Otherwise, print e.g. "%constant.42 (s32[100])".