Searched refs:hlo_profile_index_map_ (Results 1 – 3 of 3) sorted by relevance
127 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()163 const HloProfileIndexMap& hlo_profile_index_map_; variable
129 hlo_profile_index_map_(*hlo_profile_index_map), in HloExecutionProfile()131 /*count=*/hlo_profile_index_map_.total_count(), in HloExecutionProfile()136 SetCyclesTakenBy(hlo_profile_index_map_.GetProfileIndexFor(*hlo), in SetCyclesTakenBy()145 return GetCyclesTakenBy(hlo_profile_index_map_.GetProfileIndexFor(hlo)); in GetCyclesTakenBy()
245 hlo_profile_index_map_(std::move(hlo_profile_index_map)) { in Executable()247 hlo_profile_index_map_.get() == nullptr); in Executable()342 return *hlo_profile_index_map_; in hlo_profile_index_map()405 std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map_; variable