flax
b386b715 - allow custom dot_general for einsum.

Commit

1 year ago

allow custom dot_general for einsum. PiperOrigin-RevId: 627886032

References

#3884 - allow custom dot_general for einsum.

Author

a-googler

a-googler

Committer

a-googler

a-googler

Parents

Loading