Add a private property to NamedSharding called _logical_device_ids which allows you to pass a custom tile_assignment_devices() equivalent.

Add a private property to NamedSharding called _logical_device_ids which allows you to pass a custom tile_assignment_devices() equivalent. #5781