I think that should be fine (based on what I've seen from previous submissions). You can design a combined image showing the app running on those devices to highlight the responsive layout, as long as the screenshots accurately reflect the actual UI of your app submission. The key point is that the images must be real screenshots of the app as built, not mockups or concepts that differ from the submitted version.
Topic:
Community
SubTopic:
Swift Student Challenge
Tags: