import tensorflow as tf

            state[self.ITERATION_STATE_KEY] + tf.constant(1, dtype=tf.int32)
    }

  def get_params(self, state):
    """See base class."""
    params = {
        self.ADD_PARAM_KEY: 1 / tf.to_float(state[self.ITERATION_STATE_KEY])
    }
    return params, params

  def encode(self, x, encode_params):
    """See base class."""
