Searched refs:profile_counters_ (Results 1 – 4 of 4) 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()145 return PrintHloProfile(hlo_profile_printer_data_, profile_counters_.data(), in ToString()149 std::vector<int64>* mutable_profile_counters() { return &profile_counters_; } in mutable_profile_counters()151 return profile_counters_; in profile_counters()160 std::vector<int64> profile_counters_; variable
129 profile_counters_( 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()
47 profile_counters_ = new int64[static_data.profile_counters_size_](); in XlaCompiledCpuFunction()53 buffer_table_, profile_counters_); in Run()60 delete[] profile_counters_; in ~XlaCompiledCpuFunction()
204 const int64* profile_counters() const { return profile_counters_; } in profile_counters()340 int64* profile_counters_ = nullptr; variable