Support nested REE in arrow-ord `partition` function

[liamzwbao]

While working on REE optimization in arrow-json decoder, we found that the partition API cannot handle nested REE gracefully, which caused the failure described in the below comment. Fix is needed in the partition func.

Originally posted by @alamb in #9497 (comment)