Merge pull request #11 from google-ai-edge:cleanup-copyright

PiperOrigin-RevId: 913290422

Author: copybara-github

test_scripts/litert_cli.ipynb

@@ -1,5 +1,27 @@
 {
  "cells": [
+  {
+   "id": "cdbfc46c",
+   "cell_type": "raw",
+   "source": [
+    "# Copyright 2026 The LiteRT CLI Authors.\n",
+    "#\n",
+    "# Licensed under the Apache License, Version 2.0 (the \"License\");\n",
+    "# you may not use this file except in compliance with the License.\n",
+    "# You may obtain a copy of the License at\n",
+    "#\n",
+    "#     http://www.apache.org/licenses/LICENSE-2.0\n",
+    "#\n",
+    "# Unless required by applicable law or agreed to in writing, software\n",
+    "# distributed under the License is distributed on an \"AS IS\" BASIS,\n",
+    "# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
+    "# See the License for the specific language governing permissions and\n",
+    "# limitations under the License.\n",
+    "# =============================================================================="
+   ],
+   "metadata": {},
+   "execution_count": null
+  },
   {
    "id": "90c3deda",
    "cell_type": "markdown",
@@ -8,15 +30,17 @@
     "\n",
     "This notebook demonstrates how to use the `litert-cli` tool to convert a PyTorch model, quantize it and run it."
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "19270446",
    "cell_type": "markdown",
    "source": [
-    "## \ud83d\udee0\ufe0f 1. Environment Setup & Installation"
+    "## 🛠️ 1. Environment Setup \u0026 Installation"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "d34739df",
@@ -34,16 +58,16 @@
     "!sudo ./llvm.sh 18 all"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "1dd1eb34",
    "cell_type": "markdown",
    "source": [
-    "## \ud83d\udcdd 2. Prepare PyTorch Model Script"
+    "## 📝 2. Prepare PyTorch Model Script"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "5e40f66c",
@@ -53,27 +77,27 @@
     "import torch\n",
     "import torchvision\n",
     "\n",
-    "def get_model() -> torch.nn.Module:\n",
+    "def get_model() -\u003e torch.nn.Module:\n",
     "  model = torchvision.models.resnet18(\n",
     "      weights=torchvision.models.ResNet18_Weights.IMAGENET1K_V1\n",
     "  )\n",
     "  model.eval()\n",
     "  return model\n",
     "\n",
-    "def get_args() -> tuple[torch.Tensor, ...]:\n",
+    "def get_args() -\u003e tuple[torch.Tensor, ...]:\n",
     "  return (torch.randn(1, 3, 224, 224),)"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "427dca8a",
    "cell_type": "markdown",
    "source": [
-    "## \ud83d\udd04 3. Model Conversion (PyTorch -> LiteRT)"
+    "## 🔄 3. Model Conversion (PyTorch -\u003e LiteRT)"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "578cc653",
@@ -83,16 +107,16 @@
     "!litert convert resnet18.py --output resnet18"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "606efd4f",
    "cell_type": "markdown",
    "source": [
-    "## \ud83d\udcc9 4. Model Quantization"
+    "## 📉 4. Model Quantization"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "c589c711",
@@ -103,17 +127,17 @@
     "!litert quantize resnet18/resnet18.tflite --type int8_weight_only --output resnet18/resnet18_int8_weight_only.tflite"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "b3ae9ddc",
    "cell_type": "markdown",
    "source": [
-    "## \ud83d\ude80 5. Run Inference\n",
-    "### \ud83d\udda5\ufe0f 5.1 CPU Inference"
+    "## 🚀 5. Run Inference\n",
+    "### 🖥️ 5.1 CPU Inference"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "ea93a1db",
@@ -124,16 +148,16 @@
     "!litert run resnet18/resnet18_int8_dynamic.tflite --desktop --cpu --iterations 1"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "9d748475",
    "cell_type": "markdown",
    "source": [
-    "### \ud83c\udfae 5.2 GPU Inference"
+    "### 🎮 5.2 GPU Inference"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "00b026b2",
@@ -144,16 +168,16 @@
     "!litert run resnet18/resnet18_int8_dynamic.tflite --desktop --gpu --iterations 1"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "eaa972bc",
    "cell_type": "markdown",
    "source": [
-    "## \u2699\ufe0f 6. NPU Offline Compilation (AOT)"
+    "## ⚙️ 6. NPU Offline Compilation (AOT)"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "d0a61181",
@@ -163,24 +187,24 @@
     "# TIP: Only support running on Linux and it might take a few minutes, given download large SDKs from SOCs.\n",
     "# TIP: It depends on Clang, and please make sure your Clang has version `18.x.x` or above when running\n",
     "#   `clang --version`. To update, choose one of below: \n",
-    "#   a) `sudo apt update && sudo apt upgrade -y`\n",
+    "#   a) `sudo apt update \u0026\u0026 sudo apt upgrade -y`\n",
     "#   b) wget https://apt.llvm.org/llvm.sh\n",
     "#      chmod +x llvm.sh\n",
     "#      sudo ./llvm.sh 18 all\n",
     "#\n",
     "!litert compile resnet18/resnet18.tflite --target sm8750"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "2da953c9",
    "cell_type": "markdown",
    "source": [
-    "## \ud83c\udfc1 7. Benchmark in Google AI Edge Portal"
+    "## 🏁 7. Benchmark in Google AI Edge Portal"
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   },
   {
    "id": "a9a19e48",
@@ -192,15 +216,14 @@
     "!gcloud auth login\n",
     "\n",
     "# Benchmark on Google AI Edge Portal\n",
-    "# Please specify you own GCP project: --gcp-project <your-own-gcp-project-id>\n",
+    "# Please specify you own GCP project: --gcp-project \u003cyour-own-gcp-project-id\u003e\n",
     "# Or set a default environment variable: LITERT_GCP_PROJECT\n",
     "!litert benchmark model.tflite --gcp --device \"pixel 7\" --cpu --gcp-project aep-e2e-test\n",
     "!litert benchmark model.tflite --gcp --device \"pixel 7\" --gpu --gcp-project aep-e2e-test\n",
     "!litert benchmark model.tflite --gcp --devices \"pixel 7, sm-s931u1\" --gpu"
    ],
    "metadata": {},
-   "execution_count": null,
-   "outputs": []
+   "execution_count": null
   },
   {
    "id": "2685b076",
@@ -219,7 +242,8 @@
     "```\n",
     "These are not executed in this notebook as Colab does not have access to a physical Android device by default."
    ],
-   "metadata": {}
+   "metadata": {},
+   "execution_count": null
   }
  ],
  "metadata": {