From 9271534856610b1b5daedf8a050b5afea7bef2a0 Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Fri, 13 Oct 2023 19:42:51 -0700 Subject: [PATCH] Add TorchScript test file (#1167) --- source/pytorch-metadata.json | 40 ++++++++++++++++++++++++++++++++++++ test/models.json | 7 +++++++ 2 files changed, 47 insertions(+) diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json index 1861552c1d..7dc4125ac2 100755 --- 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 index ba4a795777..d028bc5949 100644 --- 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",