jax
dc446a12 - Move the TPU lowering basic_ragged_dot case (reliant on a fusion) to the

Commit

34 days ago

Move the TPU lowering basic_ragged_dot case (reliant on a fusion) to the lowering from the transpose rule where it is currently. PiperOrigin-RevId: 879132799

References

#35618 - Move the TPU lowering basic_ragged_dot case (reliant on a fusion) to the

Author

rdyro

Committer

Google-ML-Automation

Parents

c0751945

jax dc446a12 - Move the TPU lowering basic_ragged_dot case (reliant on a fusion) to the

jax
dc446a12 - Move the TPU lowering basic_ragged_dot case (reliant on a fusion) to the