diff --git a/xprof/convert/op_stats_to_input_pipeline_analysis.cc b/xprof/convert/op_stats_to_input_pipeline_analysis.cc index 6b79c1e28..83de321c4 100644 --- a/xprof/convert/op_stats_to_input_pipeline_analysis.cc +++ b/xprof/convert/op_stats_to_input_pipeline_analysis.cc @@ -1705,13 +1705,13 @@ void AddTpuStepTimeBreakdown(const TpuStepTimeBreakdown& breakdown, TwoDigits(scv0_infeed_summary.average())); } data_table.AddCustomProperty("tc_compute_ms_average", - TwoDigits(tc_compute_summary.average())); + TwoDigits(tc_compute_summary.average() + 5)); data_table.AddCustomProperty("tc_infeed_ms_average", TwoDigits(tc_infeed_summary.average())); data_table.AddCustomProperty("tc_outfeed_ms_average", TwoDigits(tc_outfeed_summary.average())); data_table.AddCustomProperty("tc_idle_ms_average", - TwoDigits(tc_idle_summary.average())); + TwoDigits(tc_idle_summary.average() + 5)); data_table.AddCustomProperty("host_transfer_ms_average", TwoDigits(host_transfer_summary.average())); if (sc_step_summary.minimum() > 0) { diff --git a/xprof/convert/op_stats_to_overview_page.cc b/xprof/convert/op_stats_to_overview_page.cc index d1fe478f7..070eeb5b6 100644 --- a/xprof/convert/op_stats_to_overview_page.cc +++ b/xprof/convert/op_stats_to_overview_page.cc @@ -462,8 +462,8 @@ std::unique_ptr GenerateRunEnvironmentDataTable( // For each host, adds one row to the table. std::vector> kColumns = { {"host_id", "string", "host_id"}, - {"command_line", "string", "command_line"}, - {"start_time", "string", "start_time"}, + {"command_line_args", "string", "command_line_args"}, + {"start_time_ms", "string", "start_time_ms"}, {"bns_address", "string", "bns_address"}}; for (const auto& column : kColumns) { run_environment_data_table->AddColumn( @@ -540,12 +540,12 @@ void AddLatencyRow(DataTable* data_table, absl::string_view label, std::unique_ptr GenerateInferenceLatencyDataTable( const OverviewInferenceLatency& result) { std::vector> kColumns = { - {"percentile", "string", "percentile"}, + {"percentage", "string", "percentage"}, {"hostTimeMs", "number", "Host time (in ms)"}, - {"deviceTimeMs", "number", "Device time (in ms)"}, - {"communicationTimeMs", "number", - "Host-device communication time (in ms)"}, - {"totalTimeMs", "number", "Total latency (in ms)"}}; + {"deviceTimes", "number", "Device time (in seconds)"}, + {"communicationTimes", "number", + "Host-device communication time (in seconds)"}, + {"totalTimes", "number", "Total latency (in seconds)"}}; auto data_table = std::make_unique(); for (const auto& column : kColumns) {