tf-models-nightly 2.15.0.dev20231120__py2.py3-none-any.whl → 2.15.0.dev20231122__py2.py3-none-any.whl

official/vision/serving/export_saved_model_lib.py

@@ -136,9 +136,20 @@ def export_inference_graph(
           type(params.task)))
 
   if add_tpu_function_alias:
+    if input_type == 'image_tensor':
+      inference_func = export_module.inference_from_image_tensors
+    elif input_type == 'image_bytes':
+      inference_func = export_module.inference_from_image_bytes
+    elif input_type == 'tf_example':
+      inference_func = export_module.inference_from_tf_example
+    else:
+      raise ValueError(
+          'add_tpu_function_alias is only allowed for input_type of:'
+          ' image_tensor, image_bytes, tf_example.'
+      )
     save_options = tf.saved_model.SaveOptions(
         function_aliases={
-            'tpu_candidate': export_module.inference_from_image_tensors,
+            'tpu_candidate': inference_func,
         }
     )
 

{tf_models_nightly-2.15.0.dev20231120.dist-info → tf_models_nightly-2.15.0.dev20231122.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.15.0.dev20231120
+Version: 2.15.0.dev20231122
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.15.0.dev20231120.dist-info → tf_models_nightly-2.15.0.dev20231122.dist-info}/RECORD

@@ -1087,7 +1087,7 @@ official/vision/serving/export_base_v2_test.py,sha256=xbUMwkre3sGoH6fXtVAGBXD5PE
 official/vision/serving/export_module_factory.py,sha256=L7unMBF_axsRhvUEizQxGKE4VwEBW82p_1YsfVFkua8,3549
 official/vision/serving/export_module_factory_test.py,sha256=Y2jI3BbbQZNLspaAgb3HovGZBj8oY9gjXt5EQCgKS3s,4605
 official/vision/serving/export_saved_model.py,sha256=fB_NNZEs2uExhZKKgvX0WLeDqvbl20-vy15dFZCyQ4Y,5758
-official/vision/serving/export_saved_model_lib.py,sha256=Fm6FAq6U-ODdVc2VZu-MoYulkW2TNnUF767zlBrFBv0,7675
+official/vision/serving/export_saved_model_lib.py,sha256=9rB9vKEyQ_FokLqqPi2eF7BkRz_k-gGEh6ZT-aSCKgw,8116
 official/vision/serving/export_saved_model_lib_test.py,sha256=kuQPLh5s--JeNq0-oTuZXIK5BZcwaMXdR1v12NNwvu8,2409
 official/vision/serving/export_saved_model_lib_v2.py,sha256=9hGuKwhTdMOX9Rq42TVdb4zuYMPsuurTDmsesEzNEyA,3710
 official/vision/serving/export_tfhub.py,sha256=ES-BDGUjhfv8u93S1RKnY-ZJw-g25NAxKAygNHpLvfI,3500
@@ -1158,9 +1158,9 @@ tensorflow_models/__init__.py,sha256=Ciz_YBke6teb6y42QyQTUBDdXJAiV7Qdu1zOoZvYiKw
 tensorflow_models/tensorflow_models_test.py,sha256=qUBLFZg7rmKkVQ3cHJVlkoid8cPqjjyc2ZiWtjQO5_o,1395
 tensorflow_models/nlp/__init__.py,sha256=3dULDpUBpDi9vljpXadq6oJrWH4y6z42Bz2d3hopYZw,807
 tensorflow_models/vision/__init__.py,sha256=4y77XkHaH8qLls3-6ta4tMp3Xj8CLbB0ihH91HsQ9z4,833
-tf_models_nightly-2.15.0.dev20231120.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.15.0.dev20231120.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.15.0.dev20231120.dist-info/METADATA,sha256=WmR0fdlC9x-cLuXel2RpuriJoJQcVH01gKySPDVkxgk,1414
-tf_models_nightly-2.15.0.dev20231120.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.15.0.dev20231120.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.15.0.dev20231120.dist-info/RECORD,,
+tf_models_nightly-2.15.0.dev20231122.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.15.0.dev20231122.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.15.0.dev20231122.dist-info/METADATA,sha256=IP6g_k2OhkkwCGExWoEVVnLH5mXFnWWBgdWU4041ni8,1414
+tf_models_nightly-2.15.0.dev20231122.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.15.0.dev20231122.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.15.0.dev20231122.dist-info/RECORD,,