[ET-VK] Minor improvements to conv2d pw and dw bounds check.

Pull Request resolved: pytorch#7815

This diff contains minor improvements to the conv2d pw and dw bounds check in the Vulkan backend for Executorch.
ghstack-source-id: 263238731
@exported-using-ghexport

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

---------

Co-authored-by: Vivek Trivedi <[email protected]>

Author: 2 people

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

@@ -41,7 +41,7 @@ void main() {
     div_by_x % out_limits.y,
     div_by_x / out_limits.y);
 
-  if (any(greaterThanEqual(pos, out_limits))) {
+  if (pos.z >= out_limits.z) {
     return;
   }
 

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

@@ -59,7 +59,7 @@ void main() {
   pos.y *= BATCH_SIZE_Y;
 
   // do not process if top pixel does not fit within the output range
-  if (any(greaterThanEqual(pos, out_limits))) {
+  if (pos.z >= out_limits.z) {
     return;
   }
 

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

@@ -44,7 +44,7 @@ void main() {
     div_by_x % out_limits.y,
     div_by_x / out_limits.y);
 
-  if (any(greaterThanEqual(pos, out_limits))) {
+  if (pos.z >= out_limits.z) {
     return;
   }
 

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

@@ -79,7 +79,7 @@ void main() {
 
   // If the top left position is out of bounds, then this invocation will have
   // no work to do.
-  if (any(greaterThanEqual(ivec3(pos[0], gpos.z), out_limits.xyz))) {
+  if (gpos.z >= out_limits.z) {
     return;
   }