flax
5f28d6a0 - Add `is_leaf` argument for `traverse_utils.flatten_dict`.

Commit

5 years ago

Add `is_leaf` argument for `traverse_utils.flatten_dict`. PiperOrigin-RevId: 363684859

References

Author

adarob

Committer

a-googler

Parents