Searched refs:computation_info (Results 1 – 2 of 2) sorted by relevance
35 for (const HloComputationInfo& computation_info : in PrintHloProfile() local37 const auto& instruction_infos = computation_info.instruction_infos(); in PrintHloProfile()50 computation_info.name(), in PrintHloProfile()51 hlo_profile_printer_data.entry_computation() == computation_info.name(), in PrintHloProfile()52 counters[computation_info.profile_index()], clock_rate_ghz); in PrintHloProfile()
87 HloComputationInfo* computation_info = in CreateHloProfilePrinterData() local90 computation_info->set_name(computation->name()); in CreateHloProfilePrinterData()91 computation_info->set_profile_index(pair.second); in CreateHloProfilePrinterData()92 computation_info->mutable_instruction_infos()->Reserve( in CreateHloProfilePrinterData()97 computation_info->add_instruction_infos(); in CreateHloProfilePrinterData()