Searched defs:report_instruction_count (Results 1 – 2 of 2) sorted by relevance
44 int64 report_instruction_count = 5; member165 explicit SpmdLogger(int64_t report_instruction_count) in SpmdLogger()
97 const HloModule& module, int64_t report_instruction_count) { in ReportBeforePartition()118 const HloModule& module, int64_t report_instruction_count) { in ReportAfterPartition()132 int64_t report_instruction_count) { in ReportMemoryUsage()