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

Use group_id as step name when no other information is available.

Browse Source 
PiperOrigin-RevId: 292262246
Change-Id: Ia7648053429dda12625067112af98c4ae04a7ba3
This commit is contained in:
Jiho Choi 2020-01-29 18:48:17 -08:00 committed by TensorFlower Gardener
parent 48bf9acb2e
commit 7b265f2af1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/profiler/utils/group_events.cc
View File

@ -98,7 +98,7 @@ std::string EventNode::GetGroupName() const {
if (auto graph_type_stat = GetContextStat(StatType::kGraphType)) { if (auto graph_type_stat = GetContextStat(StatType::kGraphType)) {
name_parts.push_back((*graph_type_stat)->str_value()); name_parts.push_back((*graph_type_stat)->str_value());
} }
int64 step_num = 0; int64 step_num = group_id_.value_or(0);
if (auto step_num_stat = GetContextStat(StatType::kStepNum)) { if (auto step_num_stat = GetContextStat(StatType::kStepNum)) {
step_num = (*step_num_stat)->int64_value(); step_num = (*step_num_stat)->int64_value();
} }