Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • q.yao's avatar
    [Feature] : Add grid_sampler TensorRT Plugin (#865) · 9d80f56a
    q.yao authored 
    * add grid sample trt support

* fix align_corners=True

* change 'intep' to 'interp', fix bugs of grid sampler, better test script

* remove unused import

* move source comment to the top of trt_grid_sampler_kernel.cu
    9d80f56a
test_tensorrt.py 14.6 KB