Build with Torch 2.7 RC1
Browse files- build/torch25-cxx11-cu118-x86_64-linux/deformable_detr/{_deformable_detr_cxy6p3o2latjs.abi3.so → _deformable_detr_32802f9_dirty.abi3.so} +2 -2
- build/torch25-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch25-cxx11-cu121-x86_64-linux/deformable_detr/{_deformable_detr_esifsbuexbtbw.abi3.so → _deformable_detr_32802f9_dirty.abi3.so} +2 -2
- build/torch25-cxx11-cu121-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch25-cxx11-cu124-x86_64-linux/deformable_detr/{_deformable_detr_cuzn3o54ku5iq.abi3.so → _deformable_detr_32802f9_dirty.abi3.so} +2 -2
- build/torch25-cxx11-cu124-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch25-cxx98-cu118-x86_64-linux/deformable_detr/{_deformable_detr_gom2c5vfrl2ic.abi3.so → _deformable_detr_32802f9_dirty.abi3.so} +2 -2
- build/torch25-cxx98-cu118-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_deformable_detr_a7sajsuqrick6.abi3.so +0 -3
- build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_tyogxwmtolvok.abi3.so +0 -3
- build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_5kxpyt5yogkv2.abi3.so +0 -3
- build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_deformable_detr_titoehueyfqjg.abi3.so +0 -3
- build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_imqt5tuqtmyt4.abi3.so +0 -3
- build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_deformable_detr_qbnaho3zp2d3o.abi3.so +0 -3
- build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_5oxft6tr6jbvu.abi3.so +0 -3
- build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_deformable_detr_po264mz2i2ffg.abi3.so +0 -3
- build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_ops.py +3 -3
- build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/__init__.py +46 -0
- build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py +9 -0
- build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/layers.py +84 -0
- build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/__init__.py +46 -0
- build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py +9 -0
- build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/layers.py +84 -0
- build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/__init__.py +46 -0
- build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so +3 -0
- build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/_ops.py +9 -0
- build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/layers.py +84 -0
build/torch25-cxx11-cu118-x86_64-linux/deformable_detr/{_deformable_detr_cxy6p3o2latjs.abi3.so → _deformable_detr_32802f9_dirty.abi3.so}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9feb360fe86da592a42cd30c59d8e7037058448a69c4e1a9bc0d144e633d4692
|
3 |
+
size 5968680
|
build/torch25-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch25-cxx11-cu121-x86_64-linux/deformable_detr/{_deformable_detr_esifsbuexbtbw.abi3.so → _deformable_detr_32802f9_dirty.abi3.so}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de4305041812e2963cac6d078031c96b838f7fb83ed3bad30399173e35346979
|
3 |
+
size 5954464
|
build/torch25-cxx11-cu121-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch25-cxx11-cu124-x86_64-linux/deformable_detr/{_deformable_detr_cuzn3o54ku5iq.abi3.so → _deformable_detr_32802f9_dirty.abi3.so}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:25d3605c8705f11792cfedf190b6d9cd6b946e23499d905cb0c380cd57a78b8a
|
3 |
+
size 5935896
|
build/torch25-cxx11-cu124-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch25-cxx98-cu118-x86_64-linux/deformable_detr/{_deformable_detr_gom2c5vfrl2ic.abi3.so → _deformable_detr_32802f9_dirty.abi3.so}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ac1ba73c8446c1ae37bc0582a5087436f9fcd310379d8815d5cf1ee0b9a8722c
|
3 |
+
size 5961616
|
build/torch25-cxx98-cu118-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:721277e6e66747917a649103120cc32c2ae47ff90847af1d14464eaefefaf921
|
3 |
+
size 5947488
|
build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_deformable_detr_a7sajsuqrick6.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:56b4c64eb7931a6f580bd5b806eae1aea43b3bb8c0f115d5d202f151974a5e7b
|
3 |
-
size 5853280
|
|
|
|
|
|
|
|
build/torch25-cxx98-cu121-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aefa04a5a3a7128695a90e966fe1912b8a50a771c9430cfabffecf062bd0f8c5
|
3 |
+
size 5928920
|
build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_tyogxwmtolvok.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:7eef07a96ddf574e5b1e07476089a62659a70faa33c82fc79987c54fecb2711f
|
3 |
-
size 5834712
|
|
|
|
|
|
|
|
build/torch25-cxx98-cu124-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7f5344b7b3595eed79c7133eba87682d012be20e8edc6d8f473df3b450f8e800
|
3 |
+
size 5968656
|
build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_5kxpyt5yogkv2.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:d1c5bb5376002363e2008eb6db64ebe0c9f6c31f9a635b7420ddfb46dce16b02
|
3 |
-
size 5870352
|
|
|
|
|
|
|
|
build/torch26-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2745cdbeddcd57ea755551b2ab19d5c4d66bd7a160dea9a08fdf14411d7e434
|
3 |
+
size 5931872
|
build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_deformable_detr_titoehueyfqjg.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:76b74d4bdbb1f562474b987fd23430d12b9f033183198f35a7dfd21fcc8ce4e1
|
3 |
-
size 5837664
|
|
|
|
|
|
|
|
build/torch26-cxx11-cu124-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:645ca7ce293c49d20b1df7f0dcff10d746ed4743760a8cdcfa304da9f688dd70
|
3 |
+
size 6059296
|
build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_imqt5tuqtmyt4.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:1acd032c2f3bc530872e0839d8bec8950b01668c913539a2e14008a1e652560f
|
3 |
-
size 5944608
|
|
|
|
|
|
|
|
build/torch26-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:13411134c60a7a179bb09816668d45c63701a87110b6d96f9eecddaf4d9df547
|
3 |
+
size 5961624
|
build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_deformable_detr_qbnaho3zp2d3o.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:b9e5074a5afdb137688e20182cf4c9f7cbb1e8a69651c08a570076aeedc8c76b
|
3 |
-
size 5863320
|
|
|
|
|
|
|
|
build/torch26-cxx98-cu118-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1cbc643254301363e3ade4212a1af3c9f93c0b61798ef75c0032c5e6bd2983c8
|
3 |
+
size 5928928
|
build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_deformable_detr_5oxft6tr6jbvu.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:bd4d0f47c165b9ce95c0328cb7a52e331e4c698746ea8e4d43c7d09c193e34bd
|
3 |
-
size 5834720
|
|
|
|
|
|
|
|
build/torch26-cxx98-cu124-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6775cd3597b48feada7b9860d916c242f04fc582b25456bf543016567fbb7850
|
3 |
+
size 6056352
|
build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_deformable_detr_po264mz2i2ffg.abi3.so
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:129844ba533ee201cd3f2bb0e17a354ee8aa35176c10896454926485acdacdac
|
3 |
-
size 5945760
|
|
|
|
|
|
|
|
build/torch26-cxx98-cu126-x86_64-linux/deformable_detr/_ops.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from . import
|
3 |
-
ops = torch.ops.
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
-
return f"
|
|
|
1 |
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
|
5 |
def add_op_namespace_prefix(op_name: str):
|
6 |
"""
|
7 |
Prefix op by namespace.
|
8 |
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/__init__.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
from . import layers
|
6 |
+
|
7 |
+
|
8 |
+
def ms_deform_attn_backward(
|
9 |
+
value: torch.Tensor,
|
10 |
+
spatial_shapes: torch.Tensor,
|
11 |
+
level_start_index: torch.Tensor,
|
12 |
+
sampling_loc: torch.Tensor,
|
13 |
+
attn_weight: torch.Tensor,
|
14 |
+
grad_output: torch.Tensor,
|
15 |
+
im2col_step: int,
|
16 |
+
) -> List[torch.Tensor]:
|
17 |
+
return ops.ms_deform_attn_backward(
|
18 |
+
value,
|
19 |
+
spatial_shapes,
|
20 |
+
level_start_index,
|
21 |
+
sampling_loc,
|
22 |
+
attn_weight,
|
23 |
+
grad_output,
|
24 |
+
im2col_step,
|
25 |
+
)
|
26 |
+
|
27 |
+
|
28 |
+
def ms_deform_attn_forward(
|
29 |
+
value: torch.Tensor,
|
30 |
+
spatial_shapes: torch.Tensor,
|
31 |
+
level_start_index: torch.Tensor,
|
32 |
+
sampling_loc: torch.Tensor,
|
33 |
+
attn_weight: torch.Tensor,
|
34 |
+
im2col_step: int,
|
35 |
+
) -> torch.Tensor:
|
36 |
+
return ops.ms_deform_attn_forward(
|
37 |
+
value,
|
38 |
+
spatial_shapes,
|
39 |
+
level_start_index,
|
40 |
+
sampling_loc,
|
41 |
+
attn_weight,
|
42 |
+
im2col_step,
|
43 |
+
)
|
44 |
+
|
45 |
+
|
46 |
+
__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]
|
build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:18b1524e7d271b75016196911d433951a6d742d2a7cc27eb5a704bbb1dd5171d
|
3 |
+
size 5968752
|
build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch27-cxx11-cu118-x86_64-linux/deformable_detr/layers.py
ADDED
@@ -0,0 +1,84 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List, Union, Tuple
|
2 |
+
|
3 |
+
from torch import Tensor
|
4 |
+
from torch.autograd import Function
|
5 |
+
from torch.autograd.function import once_differentiable
|
6 |
+
import torch.nn as nn
|
7 |
+
|
8 |
+
from ._ops import ops
|
9 |
+
|
10 |
+
|
11 |
+
class MultiScaleDeformableAttentionFunction(Function):
|
12 |
+
@staticmethod
|
13 |
+
def forward(
|
14 |
+
context,
|
15 |
+
value: Tensor,
|
16 |
+
value_spatial_shapes: Tensor,
|
17 |
+
value_level_start_index: Tensor,
|
18 |
+
sampling_locations: Tensor,
|
19 |
+
attention_weights: Tensor,
|
20 |
+
im2col_step: int,
|
21 |
+
):
|
22 |
+
context.im2col_step = im2col_step
|
23 |
+
output = ops.ms_deform_attn_forward(
|
24 |
+
value,
|
25 |
+
value_spatial_shapes,
|
26 |
+
value_level_start_index,
|
27 |
+
sampling_locations,
|
28 |
+
attention_weights,
|
29 |
+
context.im2col_step,
|
30 |
+
)
|
31 |
+
context.save_for_backward(
|
32 |
+
value,
|
33 |
+
value_spatial_shapes,
|
34 |
+
value_level_start_index,
|
35 |
+
sampling_locations,
|
36 |
+
attention_weights,
|
37 |
+
)
|
38 |
+
return output
|
39 |
+
|
40 |
+
@staticmethod
|
41 |
+
@once_differentiable
|
42 |
+
def backward(context, grad_output):
|
43 |
+
(
|
44 |
+
value,
|
45 |
+
value_spatial_shapes,
|
46 |
+
value_level_start_index,
|
47 |
+
sampling_locations,
|
48 |
+
attention_weights,
|
49 |
+
) = context.saved_tensors
|
50 |
+
grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
|
51 |
+
value,
|
52 |
+
value_spatial_shapes,
|
53 |
+
value_level_start_index,
|
54 |
+
sampling_locations,
|
55 |
+
attention_weights,
|
56 |
+
grad_output,
|
57 |
+
context.im2col_step,
|
58 |
+
)
|
59 |
+
|
60 |
+
return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
|
61 |
+
|
62 |
+
|
63 |
+
class MultiScaleDeformableAttention(nn.Module):
|
64 |
+
def forward(
|
65 |
+
self,
|
66 |
+
value: Tensor,
|
67 |
+
value_spatial_shapes: Tensor,
|
68 |
+
value_spatial_shapes_list: List[Tuple],
|
69 |
+
level_start_index: Tensor,
|
70 |
+
sampling_locations: Tensor,
|
71 |
+
attention_weights: Tensor,
|
72 |
+
im2col_step: int,
|
73 |
+
):
|
74 |
+
return MultiScaleDeformableAttentionFunction.apply(
|
75 |
+
value,
|
76 |
+
value_spatial_shapes,
|
77 |
+
level_start_index,
|
78 |
+
sampling_locations,
|
79 |
+
attention_weights,
|
80 |
+
im2col_step,
|
81 |
+
)
|
82 |
+
|
83 |
+
|
84 |
+
__all__ = ["MultiScaleDeformableAttention"]
|
build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/__init__.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
from . import layers
|
6 |
+
|
7 |
+
|
8 |
+
def ms_deform_attn_backward(
|
9 |
+
value: torch.Tensor,
|
10 |
+
spatial_shapes: torch.Tensor,
|
11 |
+
level_start_index: torch.Tensor,
|
12 |
+
sampling_loc: torch.Tensor,
|
13 |
+
attn_weight: torch.Tensor,
|
14 |
+
grad_output: torch.Tensor,
|
15 |
+
im2col_step: int,
|
16 |
+
) -> List[torch.Tensor]:
|
17 |
+
return ops.ms_deform_attn_backward(
|
18 |
+
value,
|
19 |
+
spatial_shapes,
|
20 |
+
level_start_index,
|
21 |
+
sampling_loc,
|
22 |
+
attn_weight,
|
23 |
+
grad_output,
|
24 |
+
im2col_step,
|
25 |
+
)
|
26 |
+
|
27 |
+
|
28 |
+
def ms_deform_attn_forward(
|
29 |
+
value: torch.Tensor,
|
30 |
+
spatial_shapes: torch.Tensor,
|
31 |
+
level_start_index: torch.Tensor,
|
32 |
+
sampling_loc: torch.Tensor,
|
33 |
+
attn_weight: torch.Tensor,
|
34 |
+
im2col_step: int,
|
35 |
+
) -> torch.Tensor:
|
36 |
+
return ops.ms_deform_attn_forward(
|
37 |
+
value,
|
38 |
+
spatial_shapes,
|
39 |
+
level_start_index,
|
40 |
+
sampling_loc,
|
41 |
+
attn_weight,
|
42 |
+
im2col_step,
|
43 |
+
)
|
44 |
+
|
45 |
+
|
46 |
+
__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]
|
build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9520e53aea7170877501f156ede53454eaaa5e1c5485610e5272813e047b8775
|
3 |
+
size 6059328
|
build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch27-cxx11-cu126-x86_64-linux/deformable_detr/layers.py
ADDED
@@ -0,0 +1,84 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List, Union, Tuple
|
2 |
+
|
3 |
+
from torch import Tensor
|
4 |
+
from torch.autograd import Function
|
5 |
+
from torch.autograd.function import once_differentiable
|
6 |
+
import torch.nn as nn
|
7 |
+
|
8 |
+
from ._ops import ops
|
9 |
+
|
10 |
+
|
11 |
+
class MultiScaleDeformableAttentionFunction(Function):
|
12 |
+
@staticmethod
|
13 |
+
def forward(
|
14 |
+
context,
|
15 |
+
value: Tensor,
|
16 |
+
value_spatial_shapes: Tensor,
|
17 |
+
value_level_start_index: Tensor,
|
18 |
+
sampling_locations: Tensor,
|
19 |
+
attention_weights: Tensor,
|
20 |
+
im2col_step: int,
|
21 |
+
):
|
22 |
+
context.im2col_step = im2col_step
|
23 |
+
output = ops.ms_deform_attn_forward(
|
24 |
+
value,
|
25 |
+
value_spatial_shapes,
|
26 |
+
value_level_start_index,
|
27 |
+
sampling_locations,
|
28 |
+
attention_weights,
|
29 |
+
context.im2col_step,
|
30 |
+
)
|
31 |
+
context.save_for_backward(
|
32 |
+
value,
|
33 |
+
value_spatial_shapes,
|
34 |
+
value_level_start_index,
|
35 |
+
sampling_locations,
|
36 |
+
attention_weights,
|
37 |
+
)
|
38 |
+
return output
|
39 |
+
|
40 |
+
@staticmethod
|
41 |
+
@once_differentiable
|
42 |
+
def backward(context, grad_output):
|
43 |
+
(
|
44 |
+
value,
|
45 |
+
value_spatial_shapes,
|
46 |
+
value_level_start_index,
|
47 |
+
sampling_locations,
|
48 |
+
attention_weights,
|
49 |
+
) = context.saved_tensors
|
50 |
+
grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
|
51 |
+
value,
|
52 |
+
value_spatial_shapes,
|
53 |
+
value_level_start_index,
|
54 |
+
sampling_locations,
|
55 |
+
attention_weights,
|
56 |
+
grad_output,
|
57 |
+
context.im2col_step,
|
58 |
+
)
|
59 |
+
|
60 |
+
return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
|
61 |
+
|
62 |
+
|
63 |
+
class MultiScaleDeformableAttention(nn.Module):
|
64 |
+
def forward(
|
65 |
+
self,
|
66 |
+
value: Tensor,
|
67 |
+
value_spatial_shapes: Tensor,
|
68 |
+
value_spatial_shapes_list: List[Tuple],
|
69 |
+
level_start_index: Tensor,
|
70 |
+
sampling_locations: Tensor,
|
71 |
+
attention_weights: Tensor,
|
72 |
+
im2col_step: int,
|
73 |
+
):
|
74 |
+
return MultiScaleDeformableAttentionFunction.apply(
|
75 |
+
value,
|
76 |
+
value_spatial_shapes,
|
77 |
+
level_start_index,
|
78 |
+
sampling_locations,
|
79 |
+
attention_weights,
|
80 |
+
im2col_step,
|
81 |
+
)
|
82 |
+
|
83 |
+
|
84 |
+
__all__ = ["MultiScaleDeformableAttention"]
|
build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/__init__.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
from . import layers
|
6 |
+
|
7 |
+
|
8 |
+
def ms_deform_attn_backward(
|
9 |
+
value: torch.Tensor,
|
10 |
+
spatial_shapes: torch.Tensor,
|
11 |
+
level_start_index: torch.Tensor,
|
12 |
+
sampling_loc: torch.Tensor,
|
13 |
+
attn_weight: torch.Tensor,
|
14 |
+
grad_output: torch.Tensor,
|
15 |
+
im2col_step: int,
|
16 |
+
) -> List[torch.Tensor]:
|
17 |
+
return ops.ms_deform_attn_backward(
|
18 |
+
value,
|
19 |
+
spatial_shapes,
|
20 |
+
level_start_index,
|
21 |
+
sampling_loc,
|
22 |
+
attn_weight,
|
23 |
+
grad_output,
|
24 |
+
im2col_step,
|
25 |
+
)
|
26 |
+
|
27 |
+
|
28 |
+
def ms_deform_attn_forward(
|
29 |
+
value: torch.Tensor,
|
30 |
+
spatial_shapes: torch.Tensor,
|
31 |
+
level_start_index: torch.Tensor,
|
32 |
+
sampling_loc: torch.Tensor,
|
33 |
+
attn_weight: torch.Tensor,
|
34 |
+
im2col_step: int,
|
35 |
+
) -> torch.Tensor:
|
36 |
+
return ops.ms_deform_attn_forward(
|
37 |
+
value,
|
38 |
+
spatial_shapes,
|
39 |
+
level_start_index,
|
40 |
+
sampling_loc,
|
41 |
+
attn_weight,
|
42 |
+
im2col_step,
|
43 |
+
)
|
44 |
+
|
45 |
+
|
46 |
+
__all__ = ["layers", "ms_deform_attn_forward", "ms_deform_attn_backward"]
|
build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/_deformable_detr_32802f9_dirty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c9943c9838bd6ec8fe3062495dc5511f5a570a71d0b786101f71ad95e3bc9de7
|
3 |
+
size 5258408
|
build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _deformable_detr_32802f9_dirty
|
3 |
+
ops = torch.ops._deformable_detr_32802f9_dirty
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_deformable_detr_32802f9_dirty::{op_name}"
|
build/torch27-cxx11-cu128-x86_64-linux/deformable_detr/layers.py
ADDED
@@ -0,0 +1,84 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import List, Union, Tuple
|
2 |
+
|
3 |
+
from torch import Tensor
|
4 |
+
from torch.autograd import Function
|
5 |
+
from torch.autograd.function import once_differentiable
|
6 |
+
import torch.nn as nn
|
7 |
+
|
8 |
+
from ._ops import ops
|
9 |
+
|
10 |
+
|
11 |
+
class MultiScaleDeformableAttentionFunction(Function):
|
12 |
+
@staticmethod
|
13 |
+
def forward(
|
14 |
+
context,
|
15 |
+
value: Tensor,
|
16 |
+
value_spatial_shapes: Tensor,
|
17 |
+
value_level_start_index: Tensor,
|
18 |
+
sampling_locations: Tensor,
|
19 |
+
attention_weights: Tensor,
|
20 |
+
im2col_step: int,
|
21 |
+
):
|
22 |
+
context.im2col_step = im2col_step
|
23 |
+
output = ops.ms_deform_attn_forward(
|
24 |
+
value,
|
25 |
+
value_spatial_shapes,
|
26 |
+
value_level_start_index,
|
27 |
+
sampling_locations,
|
28 |
+
attention_weights,
|
29 |
+
context.im2col_step,
|
30 |
+
)
|
31 |
+
context.save_for_backward(
|
32 |
+
value,
|
33 |
+
value_spatial_shapes,
|
34 |
+
value_level_start_index,
|
35 |
+
sampling_locations,
|
36 |
+
attention_weights,
|
37 |
+
)
|
38 |
+
return output
|
39 |
+
|
40 |
+
@staticmethod
|
41 |
+
@once_differentiable
|
42 |
+
def backward(context, grad_output):
|
43 |
+
(
|
44 |
+
value,
|
45 |
+
value_spatial_shapes,
|
46 |
+
value_level_start_index,
|
47 |
+
sampling_locations,
|
48 |
+
attention_weights,
|
49 |
+
) = context.saved_tensors
|
50 |
+
grad_value, grad_sampling_loc, grad_attn_weight = ops.ms_deform_attn_backward(
|
51 |
+
value,
|
52 |
+
value_spatial_shapes,
|
53 |
+
value_level_start_index,
|
54 |
+
sampling_locations,
|
55 |
+
attention_weights,
|
56 |
+
grad_output,
|
57 |
+
context.im2col_step,
|
58 |
+
)
|
59 |
+
|
60 |
+
return grad_value, None, None, grad_sampling_loc, grad_attn_weight, None
|
61 |
+
|
62 |
+
|
63 |
+
class MultiScaleDeformableAttention(nn.Module):
|
64 |
+
def forward(
|
65 |
+
self,
|
66 |
+
value: Tensor,
|
67 |
+
value_spatial_shapes: Tensor,
|
68 |
+
value_spatial_shapes_list: List[Tuple],
|
69 |
+
level_start_index: Tensor,
|
70 |
+
sampling_locations: Tensor,
|
71 |
+
attention_weights: Tensor,
|
72 |
+
im2col_step: int,
|
73 |
+
):
|
74 |
+
return MultiScaleDeformableAttentionFunction.apply(
|
75 |
+
value,
|
76 |
+
value_spatial_shapes,
|
77 |
+
level_start_index,
|
78 |
+
sampling_locations,
|
79 |
+
attention_weights,
|
80 |
+
im2col_step,
|
81 |
+
)
|
82 |
+
|
83 |
+
|
84 |
+
__all__ = ["MultiScaleDeformableAttention"]
|