Update pytorch-metadata.json

lutzroeder · Apr 4, 2023 · 0f72f8c · 0f72f8c
1 parent 6346e26
commit 0f72f8c
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json
@@ -14257,6 +14257,20 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "quantized::conv1d_prepack",
+    "inputs": [
+      { "name": "weight", "type": "Tensor" },
+      { "name": "bias", "type": "Tensor", "optional": true },
+      { "name": "stride", "type": "int64[]" },
+      { "name": "padding", "type": "int64[]" },
+      { "name": "dilation", "type": "int64[]" },
+      { "name": "groups", "type": "int64" }
+    ],
+    "outputs": [
+      { "type": "__torch__.torch.classes.quantized.Conv2dPackedParamsBase" }
+    ]
+  },
   {
     "name": "quantized::conv1d_relu",
     "category": "Layer",
@@ -14458,6 +14472,20 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "quantized::conv3d_prepack",
+    "inputs": [
+      { "name": "weight", "type": "Tensor" },
+      { "name": "bias", "type": "Tensor", "optional": true },
+      { "name": "stride", "type": "int64[]" },
+      { "name": "padding", "type": "int64[]" },
+      { "name": "dilation", "type": "int64[]" },
+      { "name": "groups", "type": "int64" }
+    ],
+    "outputs": [
+      { "type": "__torch__.torch.classes.quantized.Conv3dPackedParamsBase" }
+    ]
+  },
   {
     "name": "quantized::conv3d_relu",
     "category": "Layer",
@@ -14488,6 +14516,20 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "quantized::conv_prepack",
+    "inputs": [
+      { "name": "weight", "type": "Tensor" },
+      { "name": "bias", "type": "Tensor", "optional": true },
+      { "name": "stride", "type": "int64[]" },
+      { "name": "padding", "type": "int64[]" },
+      { "name": "dilation", "type": "int64[]" },
+      { "name": "groups", "type": "int64" }
+    ],
+    "outputs": [
+      { "type": "__torch__.torch.classes.quantized.Conv2dPackedParamsBase" }
+    ]
+  },
   {
     "name": "quantized::conv_transpose1d_prepack",
     "inputs": [