• 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
affine_grid_op.h 7.4 KB