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

Prelu batching rule (forward + backward) #609

Merged
merged 2 commits into from
Apr 5, 2022
Merged

Prelu batching rule (forward + backward) #609

merged 2 commits into from
Apr 5, 2022

Conversation

samdow
Copy link
Contributor

@samdow samdow commented Mar 21, 2022

Adds the batching rule for prelu. Will need to wait on merging of prelu opinfo fix in order to get rid of the last commit that does a hacky fix so that the weight is passed as a positional arg and is therefore batchable

@samdow samdow force-pushed the prelu branch 2 times, most recently from 5c28ee4 to e8c16c8 Compare March 30, 2022 15:43
@samdow samdow marked this pull request as ready for review March 30, 2022 18:39
@samdow samdow force-pushed the prelu branch 2 times, most recently from 531dc00 to 5e8bb19 Compare March 31, 2022 19:20
@samdow samdow merged commit aa2cd89 into main Apr 5, 2022
@samdow samdow deleted the prelu branch April 5, 2022 20:57
samdow pushed a commit that referenced this pull request Apr 6, 2022
* prelu forward rule

* prelu backward rule
zou3519 pushed a commit that referenced this pull request Apr 6, 2022
* prelu forward rule

* prelu backward rule
zou3519 pushed a commit to zou3519/pytorch that referenced this pull request Jul 20, 2022
bigfootjon pushed a commit to pytorch/pytorch that referenced this pull request Jul 21, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants