feat: generalize reduce_slice_fusion#1689

avik-pal merged 1 commit intomainEnzymeAD/Enzyme-JAX:mainfrom

ap/reduce_slice_fusion_improveEnzymeAD/Enzyme-JAX:ap/reduce_slice_fusion_improveCopy head branch name to clipboard