1. 25 8月, 2020 1 次提交
    • W
      【2.0 API】Enhance affine grid operator (#26385) · a065a242
      whs 提交于
      * Enhance affine grid operator:
      1. Add cuda kernel
      2. Add align corners options
      test=develop
      
      * Move new affine_grid api to functional
      test=develop
      
      * Add CUDA kernel for affine_grid.
      test=develop
      
      * Add more unitest for grid sample API
      test=develop
      a065a242
  2. 06 1月, 2020 1 次提交
    • Z
      all cases use large shape (#22065) · 85ba5275
      zhupengyang 提交于
      affine_channel, affine_grid, bilinear_interp, bilinear_tensor_product,
      clip, crop, crop_tensor, cumsum, data_norm, expand, expand_as
      85ba5275
  3. 18 12月, 2019 1 次提交
  4. 13 12月, 2019 1 次提交
  5. 02 11月, 2018 1 次提交
    • W
      Add affine grid generator op (#12238) · 0c319e0b
      whs 提交于
      * Add affine grid generator.
      
      * fix ffine grid.
      
      * Add unitest.
      
      * Add CPU kernel and fix unitest.
      
      * Fix CPU kernel.
      
      * Refine code.
      test=develop
      
      * Fix python api.
      test=develop
      
      * Update python api.
      test=develop
      
      * Fix comment.
      test=develop
      
      * Rename affine_grid_generator to affine_grid and enhence unitest.
      test=develop
      
      * Fix unitest.
      test=develop
      0c319e0b