Searched refs:GetProfileIndexFor (Results 1 – 2 of 2) sorted by relevance
49 size_t GetProfileIndexFor(const HloInstruction& instruction) const { in GetProfileIndexFor() function53 size_t GetProfileIndexFor(const HloComputation& computation) const { in GetProfileIndexFor() function57 size_t GetProfileIndexFor(const string& key) const { in GetProfileIndexFor() function122 return profile_counters_[hlo_profile_index_map_.GetProfileIndexFor( in total_cycles_executed()129 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(computation)] = in set_total_cycles_executed()135 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(metric)] = in set_extra_metrics()
109 hlo_profile_index_map.GetProfileIndexFor(*hlo)); in CreateHloProfilePrinterData()135 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(*hlo)] = in SetCyclesTakenBy()140 return profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(hlo)]; in GetCyclesTakenBy()