message: Add a regression test for
  Was already fixed by
diff: @@ -35,7 +35,10 @@ from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.framework import test_util
  +from tensorflow.python.keras.engine import input_layer
  from tensorflow.python.keras.engine import sequential
  +from tensorflow.python.keras.engine import training as training_lib
  +from tensorflow.python.keras.layers import convolutional
  from tensorflow.python.keras.layers import core
  from tensorflow.python.keras.optimizer_v2 import adam
  from tensorflow.python.lib.io import file_io
  @@ -1439,6 +1442,17 @@ class LoadTest(test.TestCase, parameterized.TestCase):
  loaded._default_save_signature(model_input)
  loaded.signatures["serving_default"](**model_input)
  + def test_functional_model_with_conv(self, cycles):
  + x = input_layer.Input(name="x", shape=(None, None, 3), dtype=dtypes.float32)
  + conved = convolutional.Conv2D(filters=3, kernel_size=3, dilation_rate=2)(x)
  + model = training_lib.Model([x], conved)
  + model_input = array_ops.ones((1, 10, 10, 3))
  + initial_output = model.predict([model_input])
  + model = self.cycle(model, cycles)
  + self.assertAllClose(
  + [initial_output],
  + list(model.signatures["serving_default"](model_input).values()))
  +
  class SingleCycleTests(test.TestCase, parameterized.TestCase):
  