xla
5bbe5c82 - [Pallas] Improve segment_ids API UX (#7037)

Commit

1 year ago

[Pallas] Improve segment_ids API UX (#7037) Summary: Adds comment for the shape of the segment_ids. Test Plan: Skip CI.

References

Author

alanwaketan

Parents