Change KerasTensor.set_shape back to mutate the nested TensorSpec in-place.
Change KerasTensor.set_shape back to mutate the nested TensorSpec in-place.

PiperOrigin-RevId: 404778599

keras-team/keras,keras-team/keras