H1B transfer已经approved，在现在公司离职前回国旅游应该带哪个I-797？

我目前的情况是在A公司正常上班。B公司的H1B transfer已经approve，马上就拿到新的I-797了。现在有H1B visa stamp，上面写的是A公司的case。
目前计划在开始新工作前回国。有个问题是我回美的时候，我应该展示哪个I-797？ 我回美的时候应该还没真正离职，回美国以后一周内就会入职新的公司。

求助我这种情况应该展示哪个I-797？
1）有人说可以用A公司的I-797，因为你还在A公司，但是这样B公司的I-797就没有激活（I-94什么的）
2）有人说可以用B公司的I-797，这样I-94就是新的过期时间。

有点糊涂我应该怎么回美国？

非常感谢有经验的人回答！