Add support for axis_name
and axis_index_groups
to lax.ragged_all_to_all
#25660
Open
copybara-service[bot] wants to merge 1 commit intomainfrom test_708938150
+68-23
axis_name
and axis_index_groups
to lax.ragged_all_to_all
#25660