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

Update stats_calculator to fix csv format error in node name

Browse Source 
PiperOrigin-RevId: 294814694
Change-Id: I2f12f1031d20de4b104e90783ae0c73d54289c7b
This commit is contained in:
Dayeong Lee 2020-02-12 19:19:23 -08:00 committed by TensorFlower Gardener
parent af458ea2ed
commit 8cad6f59e1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/util/stats_calculator.cc
View File

@ -81,10 +81,12 @@ std::string StatsCalculator::ColumnString(const Detail& detail,
std::stringstream stream; std::stringstream stream;
if (options_.format_as_csv) { if (options_.format_as_csv) {
std::string name(detail.name);
std::replace(name.begin(), name.end(), ',', '\t');
stream << detail.type << ", " << start_ms << ", " << first_time_ms << ", " stream << detail.type << ", " << start_ms << ", " << first_time_ms << ", "
<< avg_time_ms << ", " << percentage << "%, " << cdf_percentage << avg_time_ms << ", " << percentage << "%, " << cdf_percentage
<< "%, " << detail.mem_used.newest() / 1000.0 << ", " << times_called << "%, " << detail.mem_used.newest() / 1000.0 << ", " << times_called
<< ", " << detail.name; << ", " << name;
} else { } else {
InitField(stream, 24) << detail.type; InitField(stream, 24) << detail.type;
InitField(stream, 17) << start_ms; InitField(stream, 17) << start_ms;