Searched refs:report_instruction_count (Results 1 – 2 of 2) sorted by relevance
43 int64 report_instruction_count = 5; member154 explicit SpmdLogger(int64 report_instruction_count) in SpmdLogger() argument155 : report_instruction_count_(report_instruction_count) {} in SpmdLogger()157 int64 report_instruction_count);159 int64 report_instruction_count);171 int64 report_instruction_count);
97 const HloModule& module, int64 report_instruction_count) { in ReportBeforePartition() argument108 report_instruction_count)); in ReportBeforePartition()113 report_instruction_count)); in ReportBeforePartition()118 const HloModule& module, int64 report_instruction_count) { in ReportAfterPartition() argument125 report_instruction_count)); in ReportAfterPartition()131 const HloModule& module, const F& filter, int64 report_instruction_count) { in ReportMemoryUsage() argument158 i < std::min<int64>(report_instruction_count, insts->size()); ++i) { in ReportMemoryUsage()3655 *module, options_.report_instruction_count)); in Run()3672 SpmdLogger logger(options_.report_instruction_count); in Run()3717 *module, options_.report_instruction_count)); in Run()