You're overthinking this.
I-94 will match name on passport, that's correct.
If AOS forms have married name, that's also OK, as you would have included marriage certificate proving how the maiden and married names are related.
They can mismatch, that's OK. In ideal world, I always advice to only change the name during N-400 to avoid mismatch on different IDs and simplify immigration. But in real world people want to use their married names as soon as they get married and that's their right.