Regenerate Colab notebooks using new nightly package index page. (#10603)
Follow-up to https://github.com/iree-org/iree/pull/10597 (see
https://github.com/iree-org/iree/issues/10479)
I considered pruning the diffs to just what changed with the index page,
but it's worthwhile to fully update the notebooks once in a while.
skip-ci
diff --git a/samples/colab/tensorflow_hub_import.ipynb b/samples/colab/tensorflow_hub_import.ipynb
index 9ffbf7a..047b0bd 100644
--- a/samples/colab/tensorflow_hub_import.ipynb
+++ b/samples/colab/tensorflow_hub_import.ipynb
@@ -77,7 +77,7 @@
},
"source": [
"%%capture\n",
- "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://github.com/iree-org/iree/releases"
+ "!python -m pip install iree-compiler iree-runtime iree-tools-tf -f https://iree-org.github.io/iree/pip-release-links.html"
],
"execution_count": 2,
"outputs": []
@@ -89,7 +89,7 @@
"colab": {
"base_uri": "https://localhost:8080/"
},
- "outputId": "31418cc7-ecd5-4c73-c595-ff1d6caf03cc"
+ "outputId": "0e9e1cc3-c97f-4a5c-c980-a43897fc6703"
},
"source": [
"import os\n",
@@ -111,11 +111,11 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
- "TensorFlow version: 2.5.0\n",
+ "TensorFlow version: 2.8.2\n",
"Using artifacts directory '/tmp/iree/colab_artifacts'\n"
- ],
- "name": "stdout"
+ ]
}
]
},
@@ -139,7 +139,7 @@
"base_uri": "https://localhost:8080/"
},
"id": "7fd0vmnloZo9",
- "outputId": "fcb260e8-672a-479d-c428-2a93d306d4b2"
+ "outputId": "f3c075d8-0422-40b2-c9f8-bdfd865fd4c2"
},
"source": [
"#@title Download the pretrained model\n",
@@ -154,10 +154,10 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
"Downloaded model from tfhub to path: '/tmp/tfhub_modules/426589ad685896ab7954855255a52db3442cb38d'\n"
- ],
- "name": "stdout"
+ ]
}
]
},
@@ -184,7 +184,7 @@
"base_uri": "https://localhost:8080/"
},
"id": "qiO66oEYQmsd",
- "outputId": "5d7e4f7d-e77e-45f1-c37e-11e38ed05ce2"
+ "outputId": "95950642-7225-4378-f3d6-ffeb8aedbcd3"
},
"source": [
"#@title Check for serving signatures\n",
@@ -205,6 +205,7 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
"Loaded SavedModel from '/tmp/tfhub_modules/426589ad685896ab7954855255a52db3442cb38d'\n",
"Serving signatures: []\n",
@@ -213,11 +214,19 @@
"\n",
"Checking for signature_defs using saved_model_cli:\n",
"\n",
- "The given SavedModel SignatureDef contains the following input(s):\n",
- "The given SavedModel SignatureDef contains the following output(s):\n",
- "Method name is: \n"
- ],
- "name": "stdout"
+ "Traceback (most recent call last):\n",
+ " File \"/usr/local/bin/saved_model_cli\", line 8, in <module>\n",
+ " sys.exit(main())\n",
+ " File \"/usr/local/lib/python3.7/dist-packages/tensorflow/python/tools/saved_model_cli.py\", line 1260, in main\n",
+ " args.func(args)\n",
+ " File \"/usr/local/lib/python3.7/dist-packages/tensorflow/python/tools/saved_model_cli.py\", line 745, in show\n",
+ " _show_inputs_outputs(args.dir, args.tag_set, args.signature_def)\n",
+ " File \"/usr/local/lib/python3.7/dist-packages/tensorflow/python/tools/saved_model_cli.py\", line 154, in _show_inputs_outputs\n",
+ " meta_graph_def, signature_def_key)\n",
+ " File \"/usr/local/lib/python3.7/dist-packages/tensorflow/python/tools/saved_model_cli.py\", line 115, in _get_inputs_tensor_info_from_meta_graph_def\n",
+ " f'Could not find signature \"{signature_def_key}\". Please choose from: '\n",
+ "ValueError: Could not find signature \"serving_default\". Please choose from: __saved_model_init_op\n"
+ ]
}
]
},
@@ -239,7 +248,7 @@
"base_uri": "https://localhost:8080/"
},
"id": "OlDG2OuqOBGC",
- "outputId": "044bb741-a563-4fe2-b0d9-8b47bcc5f286"
+ "outputId": "8296a409-c630-4d03-c81c-58aa95cc0f77"
},
"source": [
"#@title Look up input signatures to use when exporting\n",
@@ -255,10 +264,10 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
- " inputs: TensorSpec(shape=(None, 224, 224, 3), dtype=tf.float32, name=u'inputs')\n"
- ],
- "name": "stdout"
+ " inputs: TensorSpec(shape=(None, 224, 224, 3), dtype=tf.float32, name='inputs')\n"
+ ]
}
]
},
@@ -269,7 +278,7 @@
"base_uri": "https://localhost:8080/"
},
"id": "gnb4HhMmkgiT",
- "outputId": "3318bc8e-321c-417f-d9b0-b4cc9dcca8de"
+ "outputId": "f8cf1fe0-0bc4-4c2f-9622-325c15cb923c"
},
"source": [
"#@title Re-export the model using the known signature\n",
@@ -303,6 +312,7 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
"Saved model with serving signatures to '/tmp/resaved_model'\n",
"\n",
@@ -324,8 +334,7 @@
" shape: (1, 1001)\n",
" name: StatefulPartitionedCall:0\n",
"Method name is: tensorflow/serving/predict\n"
- ],
- "name": "stdout"
+ ]
}
]
},
@@ -345,7 +354,7 @@
"base_uri": "https://localhost:8080/"
},
"id": "GLkjlHE5mdmg",
- "outputId": "d057d2dc-001c-4fe4-cf4b-9517d4cf598e"
+ "outputId": "f6a7718a-456b-4eb1-ea0f-3465e658f3c9"
},
"source": [
"#@title Import from SavedModel\n",
@@ -383,12 +392,12 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
"Saved compiled output to '/tmp/iree/colab_artifacts/mobilenet_v2.vmfb'\n",
"Saved tf_input to '/tmp/iree/colab_artifacts/mobilenet_v2_tf_input.mlir'\n",
"Saved iree_input to '/tmp/iree/colab_artifacts/mobilenet_v2_iree_input.mlir'\n"
- ],
- "name": "stdout"
+ ]
}
]
},
@@ -397,10 +406,10 @@
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
- "height": 102
+ "height": 104
},
"id": "IEJAzOb5qASI",
- "outputId": "52da0e1f-a512-4ee8-ea65-e716de9a3cf2"
+ "outputId": "d5a6ec5e-a583-47c9-b1eb-3bd81c02b50f"
},
"source": [
"#@title Download compilation artifacts\n",
@@ -422,18 +431,21 @@
"outputs": [
{
"output_type": "stream",
+ "name": "stdout",
"text": [
"Zipping '/tmp/iree/colab_artifacts' to '/tmp/mobilenet_colab_artifacts.zip' for download...\n",
" adding: mobilenet_v2.vmfb (deflated 8%)\n",
- " adding: mobilenet_v2_iree_input.mlir (deflated 46%)\n",
- " adding: mobilenet_v2_tf_input.mlir (deflated 47%)\n",
+ " adding: mobilenet_v2_tf_input.mlir (deflated 8%)\n",
+ " adding: mobilenet_v2_iree_input.mlir (deflated 8%)\n",
"Downloading the artifacts zip file...\n"
- ],
- "name": "stdout"
+ ]
},
{
"output_type": "display_data",
"data": {
+ "text/plain": [
+ "<IPython.core.display.Javascript object>"
+ ],
"application/javascript": [
"\n",
" async function download(id, filename, size) {\n",
@@ -476,30 +488,23 @@
" div.remove();\n",
" }\n",
" "
- ],
- "text/plain": [
- "<IPython.core.display.Javascript object>"
]
},
- "metadata": {
- "tags": []
- }
+ "metadata": {}
},
{
"output_type": "display_data",
"data": {
- "application/javascript": [
- "download(\"download_173f97e9-1e2d-401a-8aad-646463432451\", \"mobilenet_colab_artifacts.zip\", 59101356)"
- ],
"text/plain": [
"<IPython.core.display.Javascript object>"
+ ],
+ "application/javascript": [
+ "download(\"download_e515e31a-0819-4066-9236-710263bad3e5\", \"mobilenet_colab_artifacts.zip\", 39551293)"
]
},
- "metadata": {
- "tags": []
- }
+ "metadata": {}
}
]
}
]
-}
+}
\ No newline at end of file
