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

[RELAY,TOPI] Add scatter_nd op #6854

Merged
merged 13 commits into from
Dec 1, 2020
Merged

Commits on Nov 11, 2020

  1. [RELAY,TOPI] Add scatter_nd op

    Scatter_nd is the inverse of gather_nd and also happens to be its
    gradient. The implementation here is not optimized. There are no cpu or
    gpu specific implementations.
    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    e5f616b View commit details
    Browse the repository at this point in the history
  2. formatting

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    3aa5bde View commit details
    Browse the repository at this point in the history
  3. Fix tests

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    dff3644 View commit details
    Browse the repository at this point in the history
  4. formatting

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    4373eba View commit details
    Browse the repository at this point in the history
  5. specify types on test

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    1d9d1f9 View commit details
    Browse the repository at this point in the history
  6. Fix grad test

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    288873f View commit details
    Browse the repository at this point in the history
  7. scatter_nd cuda impl

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    60d94d7 View commit details
    Browse the repository at this point in the history
  8. cuda impl

    Tristan Konolige authored and tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    557b915 View commit details
    Browse the repository at this point in the history
  9. x86 impl

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    2431807 View commit details
    Browse the repository at this point in the history
  10. formatting

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    0593227 View commit details
    Browse the repository at this point in the history
  11. fix shape rel

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    8cf300f View commit details
    Browse the repository at this point in the history
  12. fix tests

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    9ec74c6 View commit details
    Browse the repository at this point in the history
  13. formatting

    tkonolige committed Nov 11, 2020
    Configuration menu
    Copy the full SHA
    b212a70 View commit details
    Browse the repository at this point in the history