Migrate upsample op to new layout declare using the gen_vulkan_spv.py (#3730)

Summary:
Pull Request resolved: #3730

Migrate upsample op to new layout declare using the gen_vulkan_spv.py
- layout_declare_tensor: https://www.internalfb.com/code/fbsource/[0ba7858aa186]/fbcode/executorch/backends/vulkan/runtime/api/gen_vulkan_spv.py?lines=219
- layout_declare_ubo, ubo: Uniform Buffer Object, OpenGL data structure

bypass-github-export-checks
bypass-github-pytorch-ci-checks
bypass-github-executorch-ci-checks

Reviewed By: jorgep31415

Differential Revision: D57781160

fbshipit-source-id: b881004e05721cf3a65691f7d850cec92c0e3c54

Author: Di Xu (AR)

backends/vulkan/runtime/graph/ops/glsl/upsample.glsl

@@ -13,21 +13,13 @@
 
 #define PRECISION ${PRECISION}
 
-#define VEC4_T ${texel_type(DTYPE)}
+#define VEC4_T ${texel_load_type(DTYPE, STORAGE)}
 
 layout(std430) buffer;
 
-layout(set = 0, binding = 0, ${IMAGE_FORMAT[DTYPE]}) uniform PRECISION restrict writeonly ${IMAGE_T[NDIM][DTYPE]} image_out;
-
-layout(set = 0, binding = 1) uniform PRECISION sampler3D image_in;
-
-layout(set = 0, binding = 2) uniform PRECISION restrict OutLimits {
-  ivec3 out_limits;
-};
-
-layout(set = 0, binding = 3) uniform PRECISION restrict Sizes {
-  ivec4 sizes;
-};
+${layout_declare_tensor(0, "w", "t_out", DTYPE, STORAGE)}
+${layout_declare_tensor(1, "r", "t_in", DTYPE, STORAGE)}
+${layout_declare_ubo(2, "ivec3", "out_limits")}
 
 layout(local_size_x_id = 0, local_size_y_id = 1, local_size_z_id = 2) in;
 
@@ -38,6 +30,6 @@ void main() {
     return;
   }
 
-  VEC4_T in_texel = texelFetch(image_in, pos, 0);
-  imageStore(image_out, pos, in_texel);
+  VEC4_T in_texel = texelFetch(t_in, pos, 0);
+  imageStore(t_out, pos, in_texel);
 }

backends/vulkan/runtime/graph/ops/glsl/upsample.yaml

@@ -9,6 +9,7 @@ upsample:
     NDIM: 3
     DTYPE: float
     PACKING: C_packed
+    STORAGE: texture3d
   generate_variant_forall:
     DTYPE:
       - VALUE: half

backends/vulkan/runtime/graph/ops/impl/Upsample.cpp

@@ -49,7 +49,7 @@ void add_upsample_node(
       // Inputs and Outputs
       {{out, api::MemoryAccessType::WRITE}, {arg, api::MemoryAccessType::READ}},
       // Shader params buffers
-      {t_out->texture_limits_ubo(), graph.create_params_buffer(0.5)},
+      {t_out->texture_limits_ubo()},
       // Specialization Constants
       {},
       // Resizing Logic