"megatron/vscode:/vscode.git/clone" did not exist on "6afaffa5c674643412e55564b46ed2bb4c4f6b8b"
  • vfdev's avatar
    Unified inputs for `F.rotate` (#2495) · 76662528
    vfdev authored
    * Added code for F_t.rotate with test
    - updated F.affine tests
    
    * Rotate test tolerance to 2%
    
    * Fixes failing test
    
    * Optimized _expanded_affine_grid with a single matmul op
    
    * Recoded _compute_output_size
    76662528
functional_pil.py 17.1 KB