Skip to content

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

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 #69164

Annotations

1 warning

lint_and_typecheck

succeeded Dec 23, 2024 in 1m 15s