Update pytorch-metadata.json

lutzroeder · Feb 12, 2025 · 87c6110 · 87c6110
1 parent 46873c2
commit 87c6110
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/source/pytorch-metadata.json b/source/pytorch-metadata.json
@@ -7595,6 +7595,9 @@
   {
     "name": "detectron2::nms_rotated(Tensor boxes, Tensor scores, float iou_threshold) -> Tensor"
   },
+  {
+    "name": "detectron2::roi_align_rotated_forward(Tensor input, Tensor rois, float spatial_scale, int pooled_height, int pooled_width, int sampling_ratio) -> Tensor"
+  },
   {
     "name": "__torch__.torch.classes.rnn.CellParamsBase",
     "inputs": [

diff --git a/tools/pytorch_script.py b/tools/pytorch_script.py
@@ -67,6 +67,7 @@ def _write_metadata(metadata):
     'aten::randint_like.generator_with_low_dtype(Tensor self, SymInt low, SymInt high, *, Generator? generator, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor',
     'aten::randint_like.generator_with_low_dtype_out(Tensor self, SymInt low, SymInt high, *, Generator? generator, MemoryFormat? memory_format=None, Tensor(a!) out) -> Tensor(a!)',
     'detectron2::nms_rotated(Tensor boxes, Tensor scores, float iou_threshold) -> Tensor',
+    'detectron2::roi_align_rotated_forward(Tensor input, Tensor rois, float spatial_scale, int pooled_height, int pooled_width, int sampling_ratio) -> Tensor',
     'executorch_prim::et_view.default(Tensor self, int[] size) -> (Tensor out)',
     'executorch_prim::add.Scalar(Scalar a, Scalar b) -> Scalar',
     'executorch_prim::sub.Scalar(Scalar a, Scalar b) -> Scalar',