Searched defs: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()
97 const HloModule& module, int64 report_instruction_count) { in ReportBeforePartition()118 const HloModule& module, int64 report_instruction_count) { in ReportAfterPartition()131 const HloModule& module, const F& filter, int64 report_instruction_count) { in ReportMemoryUsage()