Searched refs:stats_calculator_ (Results 1 – 4 of 4) sorted by relevance
63 return stats_calculator_->GetOutputString(); in GetOutputString()67 return stats_calculator_->GetShortSummary(); in ShortSummary()82 stats_calculator_->ComputeStatsByType( in ComputeStatsByType()88 return stats_calculator_->GetStatsByNodeType(); in GetStatsByNodeType()94 return stats_calculator_->GetStatsByMetric(title, sorting_metric, in GetStatsByMetric()98 int num_runs() const { return stats_calculator_->num_runs(); } in num_runs()102 return stats_calculator_->run_total_us(); in run_total_us()111 std::unique_ptr<StatsCalculator> stats_calculator_; variable
37 : stats_calculator_(new StatsCalculator(options)) {} in StatSummarizer()40 : stats_calculator_(new StatsCalculator(StatSummarizerOptions())) {} in StatSummarizer()204 stats_calculator_->AddNodeStats(name, op_type, node_num, start_us, in ProcessStepStats()213 stats_calculator_->UpdateRunTotalUs(curr_total_us); in ProcessStepStats()214 stats_calculator_->UpdateMemoryUsed(mem_total); in ProcessStepStats()222 for (const auto& entry : stats_calculator_->GetDetails()) { in PrintOutputs()
41 return stats_calculator_->GetOutputString(); in GetOutputString()45 return stats_calculator_->GetShortSummary(); in GetShortSummary()49 std::unique_ptr<tensorflow::StatsCalculator> stats_calculator_;
101 : stats_calculator_( in ProfileSummarizer()135 stats_calculator_->AddNodeStats(tag_string(node_name, event->tag), in ProcessProfiles()143 stats_calculator_->UpdateRunTotalUs(curr_total_us); in ProcessProfiles()