Home
last modified time | relevance | path

Searched defs:average_parameters (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/model_averaging/
H A Daveragers.py31 def average_parameters(self, params): member in ModelAverager
104 def average_parameters( member in PeriodicModelAverager
H A Dutils.py22 def average_parameters( function
H A Dhierarchical_model_averager.py159 def average_parameters( member in HierarchicalModelAverager