flax
01afb539 - fix transform purity check to avoid tree_map.

Commit

4 years ago

fix transform purity check to avoid tree_map. PiperOrigin-RevId: 399274178

References

#1578 - fix transform purity check to avoid tree_map.

Author

levskaya

levskaya

Committer

a-googler

a-googler

Parents

Loading