[tf-rmm] SME support in RMM