flax
760cf868 - Merge pull request #3804 from google:nnx-unflatten-no-copy

Commit

1 year ago

Merge pull request #3804 from google:nnx-unflatten-no-copy PiperOrigin-RevId: 621220088

References

Author

a-googler

Parents