Skip to content

[Unity][Frontend][NN] Collection of new NN operators#15642

Merged
junrushao merged 6 commits intoapache:unityfrom
jwfromm:nn_complete
Aug 30, 2023
Merged

[Unity][Frontend][NN] Collection of new NN operators#15642
junrushao merged 6 commits intoapache:unityfrom
jwfromm:nn_complete

Conversation

@jwfromm
Copy link
Copy Markdown
Contributor

@jwfromm jwfromm commented Aug 29, 2023

This PR adds a bunch of new NN operators that conform to torch semantics. I also updated a couple of existing operators like LayerNorm to be more inline with the torch equivalent. Specifically this PR adds

  • unsqueeze
  • concat
  • subtract
  • Identity
  • chunk
  • pad
    I also fixed an issue in ReshapeDataflowRewrite when operating on scalars.

Copy link
Copy Markdown
Member

@junrushao junrushao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome!

@junrushao junrushao merged commit e0501b5 into apache:unity Aug 30, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants