import tensorflow as tf

          server_test_data = [
              d._replace(updated_state=updated_state) for d in server_test_data
          ]

      return server_test_data, decode_params

    def _non_adaptive_many_to_one_encode_decode():
      """Implementation of the method for `EncodingStageInterface`."""
      server_graph = tf.Graph()
      with server_graph.as_default():
        shape = input_values[0].shape
        encode_params, decode_params = stage.get_params()
      with self.session(server_graph) as sess:
        encode_params, decode_params = self.evaluate_tf_py_list(
            [encode_params, decode_params], sess)

