Searched refs:instruction_info (Results 1 – 3 of 3) sorted by relevance
33 [&](const HloInstructionInfo& instruction_info) { in PrintHloProfile() argument34 return counters[instruction_info.profile_index()] != 0; in PrintHloProfile()47 for (const auto& instruction_info : instruction_infos) { in PrintHloProfile() local49 /*op_name=*/instruction_info.long_name(), in PrintHloProfile()50 /*short_name=*/instruction_info.short_name(), in PrintHloProfile()51 instruction_info.category(), in PrintHloProfile()52 counters[instruction_info.profile_index()], in PrintHloProfile()53 instruction_info.flop_count(), in PrintHloProfile()54 instruction_info.transcendental_count(), in PrintHloProfile()55 instruction_info.bytes_accessed(), in PrintHloProfile()[all …]
88 HloInstructionInfo* instruction_info = in CreateHloProfilePrinterData() local90 instruction_info->set_long_name(hlo->ToString()); in CreateHloProfilePrinterData()91 instruction_info->set_short_name( in CreateHloProfilePrinterData()93 instruction_info->set_category(hlo->ToCategory()); in CreateHloProfilePrinterData()94 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo)); in CreateHloProfilePrinterData()95 instruction_info->set_transcendental_count( in CreateHloProfilePrinterData()97 instruction_info->set_bytes_accessed(cost_analysis.bytes_accessed(*hlo)); in CreateHloProfilePrinterData()98 instruction_info->set_optimal_seconds( in CreateHloProfilePrinterData()100 instruction_info->set_profile_index( in CreateHloProfilePrinterData()
80 struct instruction_info struct92 static const struct instruction_info InstInfo[MAX_OPCODE] = { argument