kernels-bot commited on
Commit
8d7e01e
·
verified ·
1 Parent(s): 5c753f4

Uploaded using `kernel-builder`.

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -21,6 +21,10 @@ ms_deform_attn_forward(...)
21
  - `ms_deform_attn_forward`
22
  - `ms_deform_attn_backward`
23
 
 
 
 
 
24
  ## Benchmarks
25
 
26
  Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/deformable-detr --version 1`.
 
21
  - `ms_deform_attn_forward`
22
  - `ms_deform_attn_backward`
23
 
24
+ ## Available layers
25
+ - `MultiScaleDeformableAttentionFunction`
26
+ - `MultiScaleDeformableAttention`
27
+
28
  ## Benchmarks
29
 
30
  Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/deformable-detr --version 1`.