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

[nnx] add split/merge_inputs #4084

Merged
merged 1 commit into from
Jul 22, 2024
Merged

[nnx] add split/merge_inputs #4084

merged 1 commit into from
Jul 22, 2024

Conversation

cgarciae
Copy link
Collaborator

@cgarciae cgarciae commented Jul 16, 2024

What does this PR do?

  • Adds the split/merge_inputs internal APIs that enable lifting a broad range of functions to work with NNX graph nodes in the inputs and outputs.
  • Minor fixes to graph and extract.

@cgarciae cgarciae force-pushed the nnx-split-merge-inputs branch 5 times, most recently from 8b89bf5 to 83f3693 Compare July 22, 2024 14:52
@cgarciae cgarciae force-pushed the nnx-split-merge-inputs branch from 83f3693 to 8299312 Compare July 22, 2024 15:06
@copybara-service copybara-service bot merged commit 8a88081 into main Jul 22, 2024
18 checks passed
@copybara-service copybara-service bot deleted the nnx-split-merge-inputs branch July 22, 2024 23:26
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