Searched defs:save_stats (Results 1 – 5 of 5) sorted by relevance
142 repeated SaveStats save_stats = 15; field
201 def save_stats(self, path: str) -> None: member in MemoryTracker
55 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