Add missing matrix matrix multiplication

Summary

This adds some more overloads of Impl::AB, for matrix-matrix multiplication with diagonal matrices.