[vulkan] Use 3D textures for everything (#67647)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67647
Test Plan: Imported from OSS
Reviewed By: beback4u
Differential Revision: D32102196
Pulled By: SS-JIA
fbshipit-source-id: ded1835386a0640181f69c190a2294d298311e26
