Skip to content
Snippets Groups Projects
Select Git revision
  • 2e821ee528c24e43856ad040c86b76a68ea82d87
  • summer default
2 results

docs

  • Clone with SSH
  • Clone with HTTPS
    • user avatar
    Add masked select module (#5356)
    Xiaoyu Zhang authored and GitHub committed 
    * add masked_select module impl

* add masked_select module test

* add masked_select module test

* add masked_select module test

* fix bug

* fix comment

* fix comment

* fix comment

* init gather_nd backward

* fix backward functor bug

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
    2e821ee5
    History
    user avatar 2e821ee5
    History
    Name Last commit Last update
    ..