Function expand

pub fn expand<C: CubePrimitive>(
    context: &mut CubeContext,
    condition: ExpandElementTyped<bool>,
    then: ExpandElementTyped<C>,
    or_else: ExpandElementTyped<C>,
) -> ExpandElementTyped<C>