jax
97cd7483 - Rename out_type -> out_sharding parameter on einsum

Commit
1 year ago
Rename out_type -> out_sharding parameter on einsum PiperOrigin-RevId: 716454800
Author
Parents
Loading