Support Half/BFloat16 in split_with_sizes.#7758

swolchok merged 2 commits intomainpytorch/executorch:mainfrom

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