• 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
affine_grid_cudnn_op.cu.cc 4.3 KB