Searched refs:instruction_info (Results 1 – 3 of 3) sorted by relevance
39 instruction_infos, [&](const HloInstructionInfo& instruction_info) { in PrintHloProfile() argument40 return counters[instruction_info.profile_index()] != 0; in PrintHloProfile()54 for (const auto& instruction_info : instruction_infos) { in PrintHloProfile() local56 /*op_name=*/instruction_info.long_name(), in PrintHloProfile()57 /*short_name=*/instruction_info.short_name(), in PrintHloProfile()58 instruction_info.category(), in PrintHloProfile()59 counters[instruction_info.profile_index()], in PrintHloProfile()60 instruction_info.flop_count(), in PrintHloProfile()61 instruction_info.transcendental_count(), in PrintHloProfile()62 instruction_info.bytes_accessed(), in PrintHloProfile()[all …]
95 HloInstructionInfo* instruction_info = in CreateHloProfilePrinterData() local97 instruction_info->set_long_name(hlo->ToString()); in CreateHloProfilePrinterData()98 instruction_info->set_short_name(hlo->ToString( in CreateHloProfilePrinterData()101 instruction_info->set_category(hlo->ToCategory()); in CreateHloProfilePrinterData()102 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo)); in CreateHloProfilePrinterData()103 instruction_info->set_transcendental_count( in CreateHloProfilePrinterData()105 instruction_info->set_bytes_accessed(cost_analysis.bytes_accessed(*hlo)); in CreateHloProfilePrinterData()106 instruction_info->set_optimal_seconds( in CreateHloProfilePrinterData()108 instruction_info->set_profile_index( in CreateHloProfilePrinterData()
80 struct instruction_info struct92 static const struct instruction_info InstInfo[MAX_OPCODE] = { argument