[11/n][ET-VK] Introduce vTensor creation from external image

Nearly all metadata is initialized to null/dummy values, except those absolutely needed in the pipeline: (1) image extents, (2) logical limits.

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

[ghstack-poisoned]

Author: jorgep31415

backends/vulkan/runtime/api/containers/Tensor.cpp

@@ -281,6 +281,21 @@ vTensorStorage::vTensorStorage(
       last_access_{},
       has_copies_{false} {}
 
+vTensorStorage::vTensorStorage(
+    Context* const context,
+    const vkapi::VulkanImage& image)
+    : context_(context),
+      storage_type_{utils::kTexture2D},
+      image_extents_(
+          {image.extents().width,
+           image.extents().height,
+           image.extents().depth}),
+      buffer_length_{0},
+      buffer_offset_{0},
+      image_(image),
+      buffer_(vkapi::VulkanBuffer()),
+      last_access_{} {}
+
 vTensorStorage::vTensorStorage(
     vTensorStorage& other,
     const int64_t buffer_offset)
@@ -446,6 +461,30 @@ vTensor::vTensor(
 }
 
 // NOLINTNEXTLINE
+vTensor::vTensor(Context* const context, const vkapi::VulkanImage& image)
+    : dtype_(),
+      // Calculate tensor metadata
+      sizes_(),
+      packed_dim_(),
+      dim_order_(),
+      axis_map_(default_axis_map()),
+      strides_(),
+      numel_(),
+      padded_sizes_(),
+      unsqueezed_strides_(),
+      padded_numel_(),
+      logical_limits_(),
+      // Utility Uniform Buffers that can be passed to shaders as arguments
+      sizes_uniform_(),
+      strides_uniform_(),
+      numel_uniform_(),
+      axis_map_uniform_(),
+      logical_limits_uniform_(),
+      // Construct Tensor storage
+      storage_(context, image) {
+  set_logical_limits(storage_.image_extents_);
+}
+
 vTensor::vTensor(vTensor& other)
     : dtype_(other.dtype_),
       // Copy tensor size metadata

backends/vulkan/runtime/api/containers/Tensor.h

@@ -95,6 +95,8 @@ class vTensorStorage final {
       const vkapi::ScalarType dtype,
       const bool allocate_memory = true);
 
+  vTensorStorage(Context* const context, const vkapi::VulkanImage& image);
+
  protected:
   /*
    * This allows for creation of tensors that use the same underlying storage
@@ -185,6 +187,8 @@ class vTensor final {
 
   vTensor(const vTensor& other) = delete;
 
+  explicit vTensor(Context* context, const vkapi::VulkanImage& image);
+
   /*
    * This constructor allows for the creation of a vTensor that references the
    * same buffer resource of another vTensor, with the same sizes and strides

backends/vulkan/runtime/graph/ComputeGraph.cpp

@@ -296,6 +296,13 @@ ValueRef ComputeGraph::add_tensor(
       sizes, dtype, suggested_memory_layout(sizes), shared_object_idx);
 }
 
+ValueRef ComputeGraph::add_tensor(const vkapi::VulkanImage& image) {
+  ValueRef idx(static_cast<int>(values_.size()));
+  check_no_active_value_ptrs();
+  values_.emplace_back(api::vTensor(context(), image));
+  return idx;
+}
+
 ValueRef ComputeGraph::add_tensor_view(const ValueRef vref) {
   const vTensorPtr t = get_tensor(vref);
   ValueRef idx(static_cast<int>(values_.size()));

backends/vulkan/runtime/graph/ComputeGraph.h

@@ -466,6 +466,11 @@ class ComputeGraph final {
       const vkapi::ScalarType dtype,
       const int64_t shared_object_idx = -1);
 
+  /*
+   * Add a `api::vTensor` value to the graph with the specified image.
+   */
+  ValueRef add_tensor(const vkapi::VulkanImage& image);
+
   /*
    * Add a `api::vTensor` value to the graph with the properties of `vref`.
    */