[ET-VK][ez] Add print operator for StorageType enum

## Context

As title. Add a long overdue print operator for the `StorageType` enum, similar to the one that already exists for the `GPUMemoryLayout` enum.

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

ghstack-source-id: 261607000
Pull Request resolved: pytorch#7685

Co-authored-by: Stephen Jia <[email protected]>

Author: pytorchbot

backends/vulkan/runtime/utils/StorageUtils.h

@@ -112,6 +112,23 @@ T to_packed_dim(const GPUMemoryLayout layout) {
   return 0;
 }
 
+inline std::ostream& operator<<(
+    std::ostream& os,
+    const StorageType storage_type) {
+  switch (storage_type) {
+    case kBuffer:
+      os << "BUFFER";
+      break;
+    case kTexture3D:
+      os << "TEXTURE_3D";
+      break;
+    case kTexture2D:
+      os << "TEXTURE_2D";
+      break;
+  }
+  return os;
+}
+
 inline std::ostream& operator<<(
     std::ostream& os,
     const GPUMemoryLayout layout) {