pub fn expand<C: CubeType>(
context: &mut CubeContext,
input: ExpandElementTyped<C>,
output: ExpandElementTyped<C>,
)
pub fn expand<C: CubeType>(
context: &mut CubeContext,
input: ExpandElementTyped<C>,
output: ExpandElementTyped<C>,
)