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

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