Searched defs:param_loc (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/benchmarks/distributed/rpc/parameter_server/server/ |
| D | server.py | 201 def param_key(self, param_loc): argument
218 def process_gradient(self, gradient, param_loc): argument
235 def average(self, param_loc): argument
249 def average_gradient(server_rref, received_batch_number, param_loc, gradient): argument
303 def process_gradient(self, gradient, param_loc): argument
319 def average(self, param_loc): argument
|