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

[XLA] Augment metadata output with source-line info, as before.

Browse Source 
PiperOrigin-RevId: 168292527
This commit is contained in:
A. Unique TensorFlower 2017-09-11 15:09:42 -07:00 committed by TensorFlower Gardener
parent 3491881522
commit 3a98035fa8
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -811,13 +811,25 @@ string HloDotDumper::GetInstructionNodeLabel(const HloInstruction* instr) {
}
string HloDotDumper::GetInstructionNodeMetadata(const HloInstruction* instr) {
if (!show_metadata_ || instr->metadata().op_name().empty()) {
if (!show_metadata_) {
return "";
}
return Printf(R"(%s<br/>op_type: %s)",
HtmlLikeStringSanitize(instr->metadata().op_name()),
HtmlLikeStringSanitize(instr->metadata().op_type()));
std::vector<string> lines;
if (!instr->metadata().op_name().empty()) {
lines.push_back(HtmlLikeStringSanitize(instr->metadata().op_name()));
}
if (!instr->metadata().op_type().empty()) {
lines.push_back(Printf(
"op_type: %s", HtmlLikeStringSanitize(instr->metadata().op_type())));
}
if (!instr->metadata().source_file().empty() &&
instr->metadata().source_line() != 0) {
lines.push_back(Printf("op_type: %s", instr->metadata().source_file(),
instr->metadata().source_line()));
}
return Join(lines, "<br/>");
}
string HloDotDumper::GetInstructionNodeExtraInfo(const HloInstruction* instr) {