Validate segmented_mm inner dimensions#3572

fallintoplace wants to merge 1 commit into

ml-explore:mainml-explore/mlx:mainfrom

fallintoplace:fix-segmented-mm-shape-checkfallintoplace/mlx:fix-segmented-mm-shape-checkCopy head branch name to clipboard