Home
last modified time | relevance | path

Searched refs:hlo_profile_index_map_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.h122 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
Dhlo_execution_profile.cc128 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()
Dexecutable.h69 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