Searched refs:ReportMemoryUsage (Results 1 – 2 of 2) sorted by relevance
170 static std::string ReportMemoryUsage(const HloModule& module, const F& filter,
102 absl::StrAppend(&report, ReportMemoryUsage( in ReportBeforePartition()111 ReportMemoryUsage( in ReportBeforePartition()123 ReportMemoryUsage( in ReportAfterPartition()130 /* static */ string SpmdLogger::ReportMemoryUsage( in ReportMemoryUsage() function in xla::spmd::SpmdLogger