Remove `on_device_shape` and `logical_on_device_shape` (#5546)
* [WIP] Remove `on_device_shape` and `logical_on_device_shape`
* logical_on_device_shape -> host_output_shape
* Patch in `is_dynamic_dimension`
* Update patch with real pending change
* Update patch
* add commit hash
