Revert "Revert "Extract parallel_for_each_reduce_over_dim_output_index from argmin parallelization PoC (#9139)""#9274

swolchok merged 5 commits intomainpytorch/executorch:mainfrom

gh/swolchok/363/headpytorch/executorch:gh/swolchok/363/headCopy head branch name to clipboard