Update pytorch-metadata.json

lutzroeder · Oct 12, 2024 · e7c2b22 · e7c2b22
1 parent 0cb3c3c
commit e7c2b22
Show file tree

Hide file tree

Showing 2 changed files with 99 additions and 1 deletion.
diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json
@@ -1210,6 +1210,44 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "aten::_upsample_bilinear2d_aa",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "output_size", "type": "SymInt[2]" },
+      { "name": "align_corners", "type": "boolean" },
+      { "name": "scales_h", "type": "float32", "optional": true, "default": null },
+      { "name": "scales_w", "type": "float32", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::_upsample_bilinear2d_aa.out",
+    "inputs": [
+      { "name": "self", "type": "Tensor" },
+      { "name": "output_size", "type": "SymInt[2]" },
+      { "name": "align_corners", "type": "boolean" },
+      { "name": "scales_h", "type": "float32", "optional": true, "default": null },
+      { "name": "scales_w", "type": "float32", "optional": true, "default": null }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::_upsample_bilinear2d_aa.vec",
+    "inputs": [
+      { "name": "input", "type": "Tensor" },
+      { "name": "output_size", "type": "SymInt[]", "optional": true },
+      { "name": "align_corners", "type": "boolean" },
+      { "name": "scale_factors", "type": "float32[]", "optional": true }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
   {
     "name": "aten::_weight_norm",
     "inputs": [
@@ -12434,6 +12472,62 @@
       { "type": "Tensor" }
     ]
   },
+  {
+    "name": "aten::searchsorted.Scalar",
+    "inputs": [
+      { "name": "sorted_sequence", "type": "Tensor" },
+      { "name": "self", "type": "Scalar" },
+      { "name": "out_int32", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "right", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "side", "type": "string", "optional": true, "default": null, "kwarg_only": true },
+      { "name": "sorter", "type": "Tensor", "optional": true, "default": null, "kwarg_only": true }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::searchsorted.Scalar_out",
+    "inputs": [
+      { "name": "sorted_sequence", "type": "Tensor" },
+      { "name": "self", "type": "Scalar" },
+      { "name": "out_int32", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "right", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "side", "type": "string", "optional": true, "default": null, "kwarg_only": true },
+      { "name": "sorter", "type": "Tensor", "optional": true, "default": null, "kwarg_only": true }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::searchsorted.Tensor",
+    "inputs": [
+      { "name": "sorted_sequence", "type": "Tensor" },
+      { "name": "self", "type": "Tensor" },
+      { "name": "out_int32", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "right", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "side", "type": "string", "optional": true, "default": null, "kwarg_only": true },
+      { "name": "sorter", "type": "Tensor", "optional": true, "default": null, "kwarg_only": true }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
+  {
+    "name": "aten::searchsorted.Tensor_out",
+    "inputs": [
+      { "name": "sorted_sequence", "type": "Tensor" },
+      { "name": "self", "type": "Tensor" },
+      { "name": "out_int32", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "right", "type": "boolean", "default": false, "kwarg_only": true },
+      { "name": "side", "type": "string", "optional": true, "default": null, "kwarg_only": true },
+      { "name": "sorter", "type": "Tensor", "optional": true, "default": null, "kwarg_only": true }
+    ],
+    "outputs": [
+      { "type": "Tensor" }
+    ]
+  },
   {
     "name": "aten::select.Dimname",
     "inputs": [

diff --git a/tools/pytorch_script.py b/tools/pytorch_script.py
@@ -68,7 +68,11 @@ def _write_metadata(value):
     'aten::as_tensor.complex(complex t, *, ScalarType? dtype=None, Device? device=None) -> Tensor',
     'aten::as_tensor.float(float t, *, ScalarType? dtype=None, Device? device=None) -> Tensor',
     'aten::as_tensor.int(int t, *, ScalarType? dtype=None, Device? device=None) -> Tensor',
-    'aten::as_tensor.list(t[] data, *, ScalarType? dtype=None, Device? device=None) -> Tensor'
+    'aten::as_tensor.list(t[] data, *, ScalarType? dtype=None, Device? device=None) -> Tensor',
+    'aten::searchsorted.Tensor(Tensor sorted_sequence, Tensor self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None) -> Tensor',  # pylint: disable=line-too-long
+    'aten::searchsorted.Tensor_out(Tensor sorted_sequence, Tensor self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None, Tensor(a!) out) -> Tensor(a!)',  # pylint: disable=line-too-long
+    'aten::searchsorted.Scalar(Tensor sorted_sequence, Scalar self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None) -> Tensor',  # pylint: disable=line-too-long
+    'aten::searchsorted.Scalar_out(Tensor sorted_sequence, Scalar self, *, bool out_int32=False, bool right=False, str? side=None, Tensor? sorter=None, Tensor(a!) out) -> Tensor(a!)',  # pylint: disable=line-too-long
 ]
 
 def _parse_schemas():