Nevermind, I found what I needed.

For the Set argument, I used a case statement.

SUM(

CASE [Dimension].[Attribute].CURRENTMEMBER

WHEN [Dimension].[Attribute].[All] THEN [Dimension].[Attribute].CURRENTMEMBER.CHILDREN

ELSE [Dimension].[Attribute].CURRENTMEMBER

END,

...

)