Return arrays from `ArrayImpl._check_and_rearrange`. Build IFRT shardings with both addressable and non-addressable devices, instead of only addressable devices.#26253

Open

copybara-service[bot] wants to merge 1 commit intomainfrom test_721929080

+57-4

Commits on Feb 4, 2025

Return arrays from ArrayImpl._check_and_rearrange. Build IFRT shardings with both addressable and non-addressable devices, instead of only addressable devices.

emilyfertig
authored and
Google-ML-Automation
committed