[ET-VK][ez] Restrict batch norm operator to 4D input tensors only#16460

SS-JIA merged 2 commits intomainpytorch/executorch:mainfrom

gh/SS-JIA/391/origpytorch/executorch:gh/SS-JIA/391/origCopy head branch name to clipboard