[TORCH][MLIR] Add E2E support for aten.native_batch_norm op
#563
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Shukla-Gaurav
force-pushed
the
gaurav/native_batch_norm
branch
from
0d5273e to
5a36240
Compare
cathyzhyi
reviewed
Feb 7, 2022
ramiro050
approved these changes
Feb 7, 2022
- This commit adds support for `aten.native_batch_norm` operation. - The current implementation only supports inference mode of `aten.native_batch_norm` op. Signed-Off-By: Gaurav Shukla <gaurav@nod-labs.com>
Shukla-Gaurav
force-pushed
the
gaurav/native_batch_norm
branch
from
5a36240 to
b6338f6
Compare
cathyzhyi
approved these changes
Feb 7, 2022
|
@Shukla-Gaurav any plan to support training mode for batch norm? |
|
@byronyi I can take a look into it. Although there is some issue with the training mode, I will post a draft for it first. This may take a couple of days. |
|
@Shukla-Gaurav FYI, there is an issue #663 with regards to training and Sean had some suggestions on how to proceed in that issue. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
4 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
aten.native_batch_normoperation.aten.native_batch_normop.Signed-Off-By: Gaurav Shukla gaurav@nod-labs.com