Update stats_calculator to fix csv format error in node name

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

View File

@ -81,10 +81,12 @@ std::string StatsCalculator::ColumnString(const Detail& detail,
std::stringstream stream;
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 << ", "
<< avg_time_ms << ", " << percentage << "%, " << cdf_percentage
<< "%, " << detail.mem_used.newest() / 1000.0 << ", " << times_called
<< ", " << detail.name;
<< ", " << name;
} else {
InitField(stream, 24) << detail.type;
InitField(stream, 17) << start_ms;