1. 28 2月, 2022 1 次提交
    • L
      Grid_sampler optimization (#39751) · 2c66775b
      Lijunhui 提交于
      * init grid_sampler with mode=bilinear
      
      * solve error
      
      * rm fill constant
      
      * rm head
      
      * change block size
      
      * change block size
      
      * optimize
      
      * apply existing config
      2c66775b
  2. 20 2月, 2022 1 次提交
  3. 11 2月, 2022 1 次提交
  4. 03 12月, 2021 1 次提交
  5. 31 5月, 2021 1 次提交
  6. 29 9月, 2020 1 次提交
  7. 28 9月, 2020 1 次提交
  8. 08 9月, 2020 1 次提交
  9. 25 8月, 2020 2 次提交
  10. 24 8月, 2020 1 次提交
    • W
      【2.0 API】Add CUDA kernel and enhance options for grid_sample (#26576) · 79539cf1
      whs 提交于
      This PR enhance CPU kernel and add new CUDA kernel to make grid_sample support:
      
      - align_corners: with bool type.
      - padding mode: which can be in ['zeros', 'reflect', 'border']
      - Interpolation mode: which ca be in ['bilinear', 'nearest']
      
      The old CPU and CUDNN version only support align_corners=true, padding_mode='zeros' and interpolation_mode='bilinear'.
      
      The behavior of the new version op in default mode is compatible with the old version.
      79539cf1