flax
7eba0111 - revert is_leaf logic in _check_carry_same_references

Commit

127 days ago

revert is_leaf logic in _check_carry_same_references PiperOrigin-RevId: 799324513

References

Author

Cristian Garcia

Committer

a-googler

Parents