(self, val, n=1)
| 453 | self.count = 0 |
| 454 | |
| 455 | def update(self, val, n=1): |
| 456 | self.val = val |
| 457 | self.sum += val * n |
| 458 | self.count += n |
| 459 | self.avg = self.sum / self.count |
| 460 | |
| 461 | def all_reduce(self): |
| 462 | if self.use_accel: |
no outgoing calls
no test coverage detected