Search for direct top squark pair production in events with a Higgs or $Z$ boson, and missing transverse momentum in $\sqrt{s}=13$ TeV $pp$ collisions with the ATLAS detector