Skip to content

Add FlattenGrad registration and test.#5461

Merged
mrry merged 1 commit into
microsoft:masterfrom
mrry:flatten_grad
Oct 15, 2020
Merged

Add FlattenGrad registration and test.#5461
mrry merged 1 commit into
microsoft:masterfrom
mrry:flatten_grad

Conversation

@mrry
Copy link
Copy Markdown
Contributor

@mrry mrry commented Oct 12, 2020

Description: Add missing gradient registration for the Flatten op.

Motivation and Context

  • Adding support for training a model that uses the Flatten op.

@mrry mrry requested a review from a team as a code owner October 12, 2020 22:15
@mrry mrry requested a review from SherlockNoMad October 13, 2020 15:27
@mrry
Copy link
Copy Markdown
Contributor Author

mrry commented Oct 14, 2020

@SherlockNoMad Looks like all the tests are passing for this... could you please review? Thanks!

@SherlockNoMad SherlockNoMad added the training issues related to ONNX Runtime training; typically submitted using template label Oct 15, 2020
@mrry mrry merged commit 64f6d85 into microsoft:master Oct 15, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer

@SherlockNoMad SherlockNoMad SherlockNoMad approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

training issues related to ONNX Runtime training; typically submitted using template

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@mrry @SherlockNoMad