更新文档

Author: liuxinwei

doc/topics/frontend/tf/index.md

@@ -1,5 +1,8 @@
 # TensorFlow 前端
 
+参考：
+1. [TF1 转换为 TF2](https://www.tensorflow.org/guide/migrate/migrate_tf2?hl=zh-cn)
+
 ```{toctree}
 tf1
 pb2onnx

tests/Conv2dTransposeReshapeConcat.ipynb

@@ -730,7 +730,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "py312x",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -744,7 +744,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.1.undefined"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tests/test.ipynb

@@ -24,6 +24,51 @@
     "import set_env"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "path = \"/media/pc/data/board/arria10/lxw/tasks/tools/npuusertools/models/tests/pytorch/demo/test.pt\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import torch\n",
+    "from tvm import relay\n",
+    "m = torch.jit.load(path)\n",
+    "shape = 1, 3, 24, 40\n",
+    "mod, params = relay.frontend.pytorch.from_pytorch(m, [(\"x\", shape)], use_parser_friendly_name=False)\n",
+    "mod = relay.transform.InferType()(mod)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "fn (%x: Tensor[(1, 3, 24, 40), float32] /* ty=Tensor[(1, 3, 24, 40), float32] span=aten::_convolution_0.x:0:0 */, %aten::_convolution_0.weight: Tensor[(16, 3, 3, 3), float32] /* ty=Tensor[(16, 3, 3, 3), float32] span=aten::_convolution_0.weight:0:0 */, %aten::_convolution_0.bias: Tensor[(16), float32] /* ty=Tensor[(16), float32] span=aten::_convolution_0.bias:0:0 */, %aten::_convolution_1.weight: Tensor[(16, 32, 3, 3), float32] /* ty=Tensor[(16, 32, 3, 3), float32] span=aten::_convolution_1.weight:0:0 */, %aten::_convolution_1.bias: Tensor[(32), float32] /* ty=Tensor[(32), float32] span=aten::_convolution_1.bias:0:0 */) -> Tensor[(1, 32, 24, 40), float32] {\n",
+      "  %0 = nn.conv2d(%x, %aten::_convolution_0.weight, padding=[1, 1, 1, 1], channels=16, kernel_size=[3, 3]) /* ty=Tensor[(1, 16, 24, 40), float32] span=aten::_convolution_0:0:0 */;\n",
+      "  %1 = nn.bias_add(%0, %aten::_convolution_0.bias) /* ty=Tensor[(1, 16, 24, 40), float32] span=aten::_convolution_0:0:0 */;\n",
+      "  %2 = nn.conv2d_transpose(%1, %aten::_convolution_1.weight, channels=32, kernel_size=[3, 3], padding=[1, 1, 1, 1]) /* ty=Tensor[(1, 32, 24, 40), float32] span=aten::_convolution_1:0:0 */;\n",
+      "  nn.bias_add(%2, %aten::_convolution_1.bias) /* ty=Tensor[(1, 32, 24, 40), float32] span=aten::_convolution_1:0:0 */\n",
+      "} /* ty=fn (Tensor[(1, 3, 24, 40), float32], Tensor[(16, 3, 3, 3), float32], Tensor[(16), float32], Tensor[(16, 32, 3, 3), float32], Tensor[(32), float32]) -> Tensor[(1, 32, 24, 40), float32] */\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(mod[\"main\"])"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 2,
@@ -168,7 +213,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "tvmz",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -182,7 +227,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.4"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,

tests/yolo-test.ipynb

@@ -1030,7 +1030,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "xin",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },

tests/yolo.ipynb

@@ -1077,7 +1077,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "xin",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },