Add expected outputs in Tensorflow SavedModel guide.

PiperOrigin-RevId: 721175407
allenxie · Jan 30, 2025 · 857b9d4 · 857b9d4
1 parent 3931d31
commit 857b9d4
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/site/en/guide/saved_model.ipynb b/site/en/guide/saved_model.ipynb
@@ -639,7 +639,7 @@
       "outputs": [],
       "source": [
         "imported_with_signatures = tf.saved_model.load(module_with_signature_path)\n",
-        "list(imported_with_signatures.signatures.keys())\n"
+        "list(imported_with_signatures.signatures.keys())  # [\"serving_default\"]"
       ]
     },
     {
@@ -674,8 +674,12 @@
       },
       "outputs": [],
       "source": [
-        "imported_with_multiple_signatures = tf.saved_model.load(module_multiple_signatures_path)\n",
-        "list(imported_with_multiple_signatures.signatures.keys())"
+        "imported_with_multiple_signatures = tf.saved_model.load(\n",
+        "    module_multiple_signatures_path\n",
+        ")\n",
+        "list(\n",
+        "    imported_with_multiple_signatures.signatures.keys()\n",
+        ")  # [\"serving_default\", \"array_input\"]"
       ]
     },
     {
@@ -720,7 +724,9 @@
       "outputs": [],
       "source": [
         "imported_with_output_name = tf.saved_model.load(module_output_path)\n",
-        "imported_with_output_name.signatures['serving_default'].structured_outputs"
+        "imported_with_output_name.signatures[\n",
+        "    'serving_default'\n",
+        "].structured_outputs  # {'custom_output_name': TensorSpec(shape=<unknown>, dtype=tf.float32, name='custom_output_name')}"
       ]
     },
     {