Internal change

PiperOrigin-RevId: 468798729

official/projects/qat/vision/serving/export_saved_model.py

@@ -65,7 +65,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string(
     'params_override', '',
     'The JSON/YAML file or string which specifies the parameter to be overriden'
     ' on top of `config_file` template.')
-_BATCH_SIZSE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
+_BATCH_SIZE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
 _IMAGE_TYPE = flags.DEFINE_string(
     'input_type', 'image_tensor',
     'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.')
@@ -114,14 +114,14 @@ def main(_):
 
   module = export_module_cls(
       params=params,
-      batch_size=_BATCH_SIZSE.value,
+      batch_size=_BATCH_SIZE.value,
       input_image_size=input_image_size,
       input_type=_IMAGE_TYPE.value,
       num_channels=3)
 
   export_saved_model_lib.export_inference_graph(
       input_type=_IMAGE_TYPE.value,
-      batch_size=_BATCH_SIZSE.value,
+      batch_size=_BATCH_SIZE.value,
       input_image_size=input_image_size,
       params=params,
       checkpoint_path=_CHECKPOINT_PATH.value,

official/vision/serving/export_saved_model.py

@@ -63,7 +63,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string(
     'params_override', '',
     'The JSON/YAML file or string which specifies the parameter to be overriden'
     ' on top of `config_file` template.')
-_BATCH_SIZSE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
+_BATCH_SIZE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
 _IMAGE_TYPE = flags.DEFINE_string(
     'input_type', 'image_tensor',
     'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.')
@@ -101,7 +101,7 @@ def main(_):
 
   export_saved_model_lib.export_inference_graph(
       input_type=_IMAGE_TYPE.value,
-      batch_size=_BATCH_SIZSE.value,
+      batch_size=_BATCH_SIZE.value,
       input_image_size=[int(x) for x in _INPUT_IMAGE_SIZE.value.split(',')],
       params=params,
       checkpoint_path=_CHECKPOINT_PATH.value,