def moving_average_update(x, value, momentum): return moving_averages.assign_moving_average( x, value, momentum, zero_debias=False) # LINEAR ALGEBRA