import tensorflow as tf

        encode_params, decode_params = stage.get_params(state)
      with self.session(server_graph) as sess:
        encode_params, decode_params, state = self.evaluate_tf_py_list(
            [encode_params, decode_params, state], sess)

      client_test_data = []
      for x in input_values:
        client_graph = tf.Graph()
        with client_graph.as_default():
          encoded_x, state_update_tensors = stage.encode(x, encode_params)
        with self.session(client_graph):
          encoded_x, state_update_tensors = self.evaluate(
              [encoded_x, state_update_tensors])
          client_test_data.append(
