Add TorchScript test file (#1167)

lutzroeder · Oct 14, 2023 · 9271534 · 9271534
1 parent 387f4ef
commit 9271534
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json
@@ -6666,6 +6666,46 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "aten::le.Scalar",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "other", "type": "Scalar" }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::le.Scalar_out",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "other", "type": "Scalar" }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::le.Tensor",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "other", "type": "Tensor" }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::le.Tensor_out",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "other", "type": "Tensor" }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
   {
     "name": "aten::leaky_relu",
     "category": "Activation",

diff --git a/test/models.json b/test/models.json
@@ -4639,6 +4639,13 @@
     "format":   "TorchScript v1.5",
     "link":     "https://github.com/lutzroeder/netron/issues/546"
   },
+  {
+    "type":     "pytorch",
+    "target":   "issue1167.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/12901438/issue1167.pt.zip[issue1167.pt]",
+    "format":   "TorchScript v1.6",
+    "link":     "https://github.com/lutzroeder/netron/issues/1167"
+  },
   {
     "type":     "pytorch",
     "target":   "junction_mlp_vehicle_model.pt",