Forráskód Böngészése

Update pytorch-metadata.json

Lutz Roeder 3 éve
szülő
commit
85f7ea99e2
1 módosított fájl, 14 hozzáadás és 1 törlés
  1. 14 1
      source/pytorch-metadata.json

+ 14 - 1
source/pytorch-metadata.json

@@ -4040,7 +4040,7 @@
     ]
   },
   {
-    "name": "torch.upsample_bicubic2d",
+    "name": "torch.upsample_bicubic2d:",
     "category": "Layer",
     "inputs": [
       { "name": "self", "type": "Tensor" },
@@ -4053,6 +4053,19 @@
       { "name": "output", "type": "Tensor" }
     ]
   },
+  {
+    "name": "torch.upsample_bicubic2d:vec",
+    "category": "Layer",
+    "inputs": [
+      { "name": "input", "type": "Tensor" },
+      { "name": "output_size", "type": "int64[]" },
+      { "name": "align_corners", "type": "boolean" },
+      { "name": "scales_factors", "type": "float32[]", "optional": true }
+    ],
+    "outputs": [
+      { "name": "output", "type": "Tensor" }
+    ]
+  },
   {
     "name": "torch.upsample_linear1d:",
     "inputs": [