Internal change

PiperOrigin-RevId: 351655727

Internal change
PiperOrigin-RevId: 351655727
de319a78 · Abdullah Rashwan · A. Unique TensorFlower · fff890ec · de319a78
Commit de319a78 authored Jan 13, 2021 by Abdullah Rashwan Committed by A. Unique TensorFlower Jan 13, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/serving/export_saved_model_lib.py official/vision/beta/serving/export_saved_model_lib.py +1 -1

No files found.
--- a/official/vision/beta/serving/export_saved_model_lib.py
+++ b/official/vision/beta/serving/export_saved_model_lib.py
@@ -86,7 +86,7 @@ def export_inference_graph(input_type, batch_size, input_image_size, params,

  if input_type == 'image_tensor':
    input_signature = tf.TensorSpec(
-        shape=[batch_size, input_image_size[0], input_image_size[1], 3],
+        shape=[batch_size, None, None, 3],
        dtype=tf.uint8)
    signatures = {
        'serving_default':