Skip to content

GitLab

Explore

Sign in
Register

Primary navigation

Project

S

Swin-Transformer-Object-Detection
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Requirements
- Environments
- Terraform modules
- Incidents
- Service Desk

Snippets Groups Projects

d7a38be9

Unverified Commit d7a38be9 authored 5 years ago by Jerry Jiarui XU Committed by GitHub 5 years ago

Downloads
- Patches
- Plain Diff

add grid sampler (#2126)

* add grid sampler

* add doc

* remove comment

parent 639f934a

No related branches found

No related tags found

No related merge requests found

Changes 9

Expand all Hide whitespace changes

Inline Side-by-side

Showing

mmdet/ops/grid_sampler/__init__.py 3 additions, 0 deletions

mmdet/ops/grid_sampler/__init__.py
mmdet/ops/grid_sampler/grid_sampler.py 117 additions, 0 deletions

mmdet/ops/grid_sampler/grid_sampler.py
mmdet/ops/grid_sampler/src/cpu/grid_sampler_cpu.cpp 692 additions, 0 deletions

mmdet/ops/grid_sampler/src/cpu/grid_sampler_cpu.cpp
mmdet/ops/grid_sampler/src/cpu/grid_sampler_cpu.h 225 additions, 0 deletions

mmdet/ops/grid_sampler/src/cpu/grid_sampler_cpu.h
mmdet/ops/grid_sampler/src/cuda/grid_sampler_cuda.cu 718 additions, 0 deletions

mmdet/ops/grid_sampler/src/cuda/grid_sampler_cuda.cu
mmdet/ops/grid_sampler/src/cuda/grid_sampler_cuda.cuh 233 additions, 0 deletions

mmdet/ops/grid_sampler/src/cuda/grid_sampler_cuda.cuh
mmdet/ops/grid_sampler/src/cudnn/grid_sampler_cudnn.cpp 148 additions, 0 deletions

mmdet/ops/grid_sampler/src/cudnn/grid_sampler_cudnn.cpp
mmdet/ops/grid_sampler/src/grid_sampler.cpp 66 additions, 0 deletions

mmdet/ops/grid_sampler/src/grid_sampler.cpp
setup.py 7 additions, 0 deletions

setup.py

with 2209 additions and 0 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment