There are two non-abelian groups of order $p^3$, namely, semi-direct product of$\mathbb Z /p \mathbb Z$ x $\mathbb Z /p \mathbb Z$ by $\mathbb Z /p \mathbb Z$ and semi-direct product of $\mathbb Z /p^2 \mathbb Z$ by $\mathbb Z /p \mathbb Z$. What are the automorphism groups of these groups?