[ET-VK] Adding convenience functions in Compute graph to get PushConstantDataInfo for various attributes of a tensor.

Differential Revision: D66853502

Pull Request resolved: #7224

Author: trivedivivek

backends/vulkan/runtime/graph/ComputeGraph.h

@@ -20,6 +20,7 @@
 #include <executorch/backends/vulkan/runtime/graph/containers/SharedObject.h>
 #include <executorch/backends/vulkan/runtime/graph/containers/Value.h>
 
+#include <executorch/backends/vulkan/runtime/graph/ops/DispatchNode.h>
 #include <executorch/backends/vulkan/runtime/graph/ops/ExecuteNode.h>
 #include <executorch/backends/vulkan/runtime/graph/ops/PrepackNode.h>
 
@@ -350,6 +351,28 @@ class ComputeGraph final {
     return values_.at(idx).toTensor().logical_limits_ubo();
   }
 
+  inline PushConstantDataInfo sizes_pc_of(const ValueRef idx) const {
+    return PushConstantDataInfo(
+        values_.at(idx).toConstTensor().get_uniform_data(), api::kTensorSizes);
+  }
+
+  inline PushConstantDataInfo strides_pc_of(const ValueRef idx) const {
+    return PushConstantDataInfo(
+        values_.at(idx).toConstTensor().get_uniform_data(),
+        api::kTensorStrides);
+  }
+
+  inline PushConstantDataInfo logical_limits_pc_of(const ValueRef idx) const {
+    return PushConstantDataInfo(
+        values_.at(idx).toConstTensor().get_uniform_data(),
+        api::kTensorLogicalLimits);
+  }
+
+  inline PushConstantDataInfo numel_pc_of(const ValueRef idx) const {
+    return PushConstantDataInfo(
+        values_.at(idx).toConstTensor().get_uniform_data(), api::kTensorNumel);
+  }
+
   //
   // Scalar Value Extraction
   //