docstr: Update the stored moving average using the latest value.
docstr_tok: ["Update","the","stored","moving","average","using","the","latest","value","."]
code: def _update_model_variables_moving_average(self, trainable_variables):
      """Update the stored moving average using the latest value."""
      if self.use_ema:
          for var, average in zip(
              trainable_variables, self._model_variables_moving_average
          ):
              not_first_step = ops.not_equal(self.iterations, 0)
              momentum = (
                  ops.cast(not_first_step, var.dtype)
                  * self.ema_momentum
              )
              average.assign(momentum * average + (1 - momentum) * var)
  
imports: import re
  import warnings
  import numpy
  from keras import backend
  from keras import initializers
  from keras import ops
  from keras.optimizers.schedules import learning_rate_schedule
  from keras.saving import serialization_lib
  from keras.utils import tracking
  from keras.utils.naming import auto_name
label_desc: Use image node_id 34 for calling the BaseOptimizer obj's underlying member method code with example usage: obj._update_model_variables_moving_average(trainable_variables) without return types