Internal change

PiperOrigin-RevId: 349583222

official/vision/beta/serving/export_base.py

@@ -86,8 +86,12 @@ class ExportModule(tf.Module, metaclass=abc.ABCMeta):
           tf.map_fn(
               _decode_tf_example,
               elems=input_tensor,
+              # Height/width of the shape of input images is unspecified (None)
+              # at the time of decoding the example, but the shape will
+              # be adjusted to conform to the input layer of the model,
+              # by _run_inference_on_image_tensors() below.
               fn_output_signature=tf.TensorSpec(
-                  shape=self._input_image_size + [3], dtype=tf.uint8),
+                  shape=[None, None, 3], dtype=tf.uint8),
               dtype=tf.uint8,
               parallel_iterations=32))
       images = tf.stack(images)