xla
Remove `on_device_shape` and `logical_on_device_shape`
#5546
Merged

Loading