Searched refs:GetProfileIndexFor (Results 1 – 4 of 4) 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() function127 return profile_counters_[hlo_profile_index_map_.GetProfileIndexFor( in total_cycles_executed()134 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(computation)] = in set_total_cycles_executed()140 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(metric)] = in set_extra_metrics()
110 hlo_profile_index_map.GetProfileIndexFor(*hlo)); in CreateHloProfilePrinterData()136 SetCyclesTakenBy(hlo_profile_index_map_.GetProfileIndexFor(*hlo), in SetCyclesTakenBy()145 return GetCyclesTakenBy(hlo_profile_index_map_.GetProfileIndexFor(hlo)); in GetCyclesTakenBy()
945 profile_index_map->GetProfileIndexFor(*module->entry_computation()); in RunBackend()
3857 condition_profile_index = index_map->GetProfileIndexFor(*condition); in BuildWhileThunk()3858 body_profile_index = index_map->GetProfileIndexFor(*body); in BuildWhileThunk()3885 body_profile_index = index_map->GetProfileIndexFor(*body); in BuildForThunk()3922 profile_index = index_map->GetProfileIndexFor(*branch_computation); in BuildConditionalThunk()5834 static_cast<int64>(index_map->GetProfileIndexFor(*hlo))); in GetThunkInfo()