Searched refs:extra_metric_to_profile_idx (Results 1 – 2 of 2) sorted by relevance
58 return xla::FindOrDie(extra_metric_to_profile_idx(), key); in GetProfileIndexFor()70 return extra_metric_to_profile_idx().size(); in extra_metrics_count()87 const std::unordered_map<string, int64>& extra_metric_to_profile_idx() const { in extra_metric_to_profile_idx() function
115 for (const auto& pair : hlo_profile_index_map.extra_metric_to_profile_idx()) { in CreateHloProfilePrinterData()