6d27bc93e1eba7d5d9dfa384593371d5a98ea5a5,tensorflow/python/keras/layers/wrappers_test.py,TimeDistributedTest,test_TimeDistributed_with_ragged_input_with_batch_size,#TimeDistributedTest#,440

Before Change

    dense_data = ragged_data.to_tensor()
    output_dense = model_2.predict(dense_data, steps=1)

    output_ragged = ragged_tensor.convert_to_tensor_or_ragged_tensor(
        output_ragged, name="tensor")
    self.assertAllEqual(output_ragged.to_tensor(), output_dense)

  def test_TimeDistributed_set_static_shape(self):
    layer = keras.layers.TimeDistributed(keras.layers.Conv2D(16, (3, 3)))

After Change

    dense_data = ragged_data.to_tensor()
    output_dense = model_2.predict(dense_data, steps=1)

    output_ragged = convert_ragged_tensor_value(output_ragged)
    self.assertAllEqual(output_ragged.to_tensor(), output_dense)

  def test_TimeDistributed_set_static_shape(self):
    layer = keras.layers.TimeDistributed(keras.layers.Conv2D(16, (3, 3)))

Instances