Searched refs:profile_counters_ (Results 1 – 4 of 4) 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()150 return PrintHloProfile(hlo_profile_printer_data_, profile_counters_.data(), in ToString()154 std::vector<int64>* mutable_profile_counters() { return &profile_counters_; } in mutable_profile_counters()156 return profile_counters_; in profile_counters()167 std::vector<int64> profile_counters_; variable
130 profile_counters_( in HloExecutionProfile()141 profile_counters_[index] = cycles_taken; in SetCyclesTakenBy()149 return profile_counters_[index]; in GetCyclesTakenBy()154 for (const auto& counter : profile_counters_) { in ToProto()
50 profile_counters_ = new int64[static_data.profile_counters_size_](); in XlaCompiledCpuFunction()56 buffer_table_, profile_counters_); in Run()63 delete[] profile_counters_; in ~XlaCompiledCpuFunction()
215 const int64* profile_counters() const { return profile_counters_; } in profile_counters()373 int64* profile_counters_ = nullptr; variable