e967f1c82f1577fded52732e68f9128c98927692,opennmt/tests/encoder_test.py,EncoderTest,testParallelEncoderCombinedOutputLayer,#EncoderTest#,208

Before Change

  @test_util.run_tf1_only
  def testParallelEncoderCombinedOutputLayer(self):
    sequence_length = [4, 6, 5]
    inputs = _build_dummy_sequences(sequence_length)
    outputs, _, encoded_length = self._encodeInParallel(
        inputs,
        sequence_length=sequence_length,
        combined_output_layer_fn=lambda x: tf.layers.dense(x, 15))

After Change

  def testParallelEncoderCombinedOutputLayer(self):
    sequence_length = tf.constant([2, 5, 4], dtype=tf.int32)
    inputs = tf.zeros([3, 5, 10])
    outputs = self._encodeInParallel(
        inputs,
        sequence_length=sequence_length,
        combined_output_layer_fn=tf.keras.layers.Dense(15))

Instances