Par manque de données sur l´origine de la locution les deux sont acceptées sauf que :
Au temps, signifiait à l´origine, un ordre militaire qui ordonnait (je crois) un temps d´action (genre "gaaarde à vous) ou un jeu de musique (chauffe Marcel).
Il devait y avoir autant "Au temps soldats, Au temps cavaliers" que "Au temps pour moi" quand ils s´agit d´un ordre concernant le chef.
Voila en très grossement résumé, dans la limite brumeuse de ma mémoire.