Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ONNX][TOPI][Relay]Support dilations in pooling operators #7928

Merged

Commits on May 3, 2021

  1. change more pooling operators

    dilations -> dilation to match old field names in conv
    
    fix python interface into new relay nodes
    
    fix order of arguments
    
    update type relation for dilations
    
    change topi interface to use dilations
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    1fff4b2 View commit details
    Browse the repository at this point in the history
  2. spooky, there are two implementations! Change to 1 topi

    use generic poolnd instead of 2d implementation for topi
    
    remove old pooling topi
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    eaa50cb View commit details
    Browse the repository at this point in the history
  3. rename pool --> pool2d in topi

    change pool -> pool2d, make topi tests work now
    
    make op level 2 pass with interface changes
    
    fix dilation being hardcoded to 1
    
    proper calculation for avgs among dilations
    
    proper avg pool padding behavior
    
    change name of pool test to pool2d test
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    58515e4 View commit details
    Browse the repository at this point in the history
  4. add poolnd baseline implementation

    more fixes to edge cases for poolnd, delete old versions
    
    replace topi tests with new baseline python version
    
    clean up tests
    
    make tests more readable kind of
    
    add dilation topi tests FINALLY
    
    remove see_pool.py
    
    remove dilation from grad
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    33e9643 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    33de2fa View commit details
    Browse the repository at this point in the history
  6. rewrite tests to be more generic for relay pooling ops

    add relay dilation tests, FINALLY
    
    add some comments to testing code
    
    linting and formatting
    
    add ASF header
    
    make 10/10 for black formatting lol
    
    more appeasing the formatting gods
    
    wow
    
    add parameters to documentation
    
    fix test import
    
    Jostle CI
    
    fix more broken unit tests using old version of pool
    
    fix wrong var used for bound calc
    
    add dilation to arm tests
    
    add docstring to python make funcs
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    971f271 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    3a62f34 View commit details
    Browse the repository at this point in the history
  8. properly forward more tests to use dilations in pooling

    formatting
    
    more formatting
    
    relax constraints on test to make it pass
    
    relax more constraints
    
    fix some pytorch frontend errors
    
    fix error
    
    better test conditions
    
    jostle build
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    c300c01 View commit details
    Browse the repository at this point in the history
  9. fix padding bug with ceil mode

    jostle build
    
    cleaner pool condition
    
    remove see_pool.py again
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    7e9c1d3 View commit details
    Browse the repository at this point in the history
  10. add dilations field to onnx importer

    blacking files
    
    black file
    AndrewZhaoLuo committed May 3, 2021
    Configuration menu
    Copy the full SHA
    856a23c View commit details
    Browse the repository at this point in the history
  11. Configuration menu
    Copy the full SHA
    3d87f4f View commit details
    Browse the repository at this point in the history