“ff51737507f5280c0fd4d6644da409a3736e9675”上不存在“paddle/framework/git@gitcode.net:paddlepaddle/Paddle.git”
  • 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
test_grid_sample_function.py 4.2 KB