Searched defs:save_stats (Results 1 – 3 of 3) sorted by relevance
201 def save_stats(self, path: str) -> None: member in MemoryTracker
49 int save_stats = 0; variable
785 …auto save_stats = batch_norm_cpu_update_stats_template<scalar_t, opmath_t, InvStd>(self, running_m… in batch_norm_cpu_out() local795 …auto save_stats = batch_norm_cpu_update_stats_template<scalar_t, scalar_t, InvStd>(self, running_m… in batch_norm_cpu_out() local