pub fn unary_expand_fixed_output<F>(
context: &mut CubeContext,
input: ExpandElement,
out_item: Item,
func: F,
) -> ExpandElement
pub fn unary_expand_fixed_output<F>(
context: &mut CubeContext,
input: ExpandElement,
out_item: Item,
func: F,
) -> ExpandElement