Searched defs:report_instruction_count (Results 1 – 2 of 2) sorted by relevance
45 int64_t report_instruction_count = 5; member166 SpmdLogger(int64_t report_instruction_count, bool disabled) in SpmdLogger()
105 const HloModule& module, int64_t report_instruction_count) { in ReportBeforePartition()126 const HloModule& module, int64_t report_instruction_count) { in ReportAfterPartition()140 int64_t report_instruction_count) { in ReportMemoryUsage()