Skip to content

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all #71169

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all #71169

Annotations

1 error

FFI example

succeeded Jan 29, 2025 in 3m 11s