import tensorflow as tf

      self.assertAllEqual((1, p.decoder.beam_search.num_hyps_per_beam),
                          fetches['scores'].shape)
      self.assertAllEqual((1, 314, p.encoder.input_shape[2], 1),
                          fetches['src_frames'].shape)
      self.assertAllEqual((80, 1, 2 * p.encoder.lstm_cell_size),
                          fetches['encoder_frames'].shape)


if __name__ == '__main__':
  tf.test.main()
