[ET-VK][ez] Add back tensor dim check

Pull Request resolved: https://github.com/pytorch/executorch/pull/7938

## Context

Vulkan cannot represent higher dimensional tensors (tensors with dim > 4) at the moment, but due to some refactors implemented last year the partitioner check to avoid lowering ops that involve high dimensional tensors was accidentally removed.

This diff adds back the check, as well as a test to verify that high dimensional tensors do not get lowered.
ghstack-source-id: 262929769
@exported-using-ghexport

Differential Revision: [D68630966](https://our.internmc.facebook.com/intern/diff/D68630966/)

Co-authored-by: Stephen Jia <ssjia@meta.com>