Update on "[ET-VK] Storing positions in uint16 to instead of int in conv2d pw shader."

This diff modifies the `conv2d_pw_s1p0.glsl` shader to store positions in `uint16` instead of `int`. The changes include adding the necessary extension for explicit arithmetic types, updating the type definitions for `TILE_SIZE_X` and `TILE_SIZE_Y`, and changing the type of the `pos` array.

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

[ghstack-poisoned]

Author: trivedivivek

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

@@ -14,8 +14,8 @@
 
 #define VEC4_T ${texel_type(DTYPE)}
 
-#define TILE_SIZE_X ${TILE_SIZE_X}us
-#define TILE_SIZE_Y ${TILE_SIZE_Y}us
+#define TILE_SIZE_X uint16_t(${TILE_SIZE_X})
+#define TILE_SIZE_Y uint16_t(${TILE_SIZE_Y})
 
 #define op(X, A, B) ${OPERATOR}
 
@@ -66,8 +66,8 @@ void main() {
   // | pos[2] | pos[3] |
   // +--------+--------+
   uint16_t pos[TILE_SIZE_X * TILE_SIZE_Y * 2];
-  for (uint16_t y = 0us, i = 0us; y < TILE_SIZE_Y; ++y) {
-    for (uint16_t x = 0us; x < TILE_SIZE_X; ++x) {
+  for (uint16_t y = uint16_t(0), i = uint16_t(0); y < TILE_SIZE_Y; ++y) {
+    for (uint16_t x = uint16_t(0); x < TILE_SIZE_X; ++x) {
       pos[i * 2] = uint16_t(out_pos[0]) * TILE_SIZE_X + x;
       pos[i * 2 + 1] = uint16_t(out_pos[1]) * TILE_SIZE_Y + y;
       i++;