Searched refs:hlo_profile_index_map_ (Results 1 – 3 of 3) sorted by relevance
122 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()156 const HloProfileIndexMap& hlo_profile_index_map_; variable
128 hlo_profile_index_map_(*hlo_profile_index_map), in HloExecutionProfile()130 /*count=*/hlo_profile_index_map_.total_count(), in HloExecutionProfile()135 profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(*hlo)] = in SetCyclesTakenBy()140 return profile_counters_[hlo_profile_index_map_.GetProfileIndexFor(hlo)]; in GetCyclesTakenBy()
69 hlo_profile_index_map_(std::move(hlo_profile_index_map)) { in Executable()71 hlo_profile_index_map_.get() == nullptr); in Executable()155 return *hlo_profile_index_map_; in hlo_profile_index_map()207 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map_; variable