Searched refs:_assign_moving_average (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 427 def _assign_moving_average(self, variable, value, momentum): member in BatchNormalizationBase
484 self.moving_mean, self._assign_moving_average,
487 self.moving_variance, self._assign_moving_average,
490 mean_update = self._assign_moving_average(self.moving_mean, mean,
492 variance_update = self._assign_moving_average(self.moving_variance,
543 new_var = self._assign_moving_average(var, value, self.renorm_momentum)
544 new_weight = self._assign_moving_average(weight, weight_value,
683 var, self._assign_moving_average, (value, self.momentum),
701 return self._assign_moving_average(var, value, self.momentum)
|