The story of Gabriel implanting a reed in the sea at the moment of Shlomo's marriage to Pharaoh's daughter, from which a sandbar rose and on which Rome was built, appears in at least two Talmudic locations: Sanhedrin 21b states it explicitly in the name of Rabbi Yitzchak — 'Gabriel descended and implanted a pole in the sea, and it gradually raised up a sandbar, and on it the great city of Rome was built' — and Shabbat 56b records the identical tradition, concluding that this was the founding of the great city of Rome.
A third Talmudic locus, Pesachim 118b, echoes the same imagery in a different context: the Holy One tells Gabriel to 'rebuke the wild beast of the reeds [kaneh],' a phrase the Gemara there reads as alluding to Rome, whose origins are tied to that very reed (kaneh) implanted in the sea.
Rashi on Sanhedrin 21b is available to illuminate the Sanhedrin passage, though the passage he comments on there concerns details of kingship and the crown rather than the reed narrative itself, so the primary sources for the story's multiple locations remain Sanhedrin 21b and Shabbat 56b.