VulkanPackedContext Class — pytorch Architecture

Source Code

aten/src/ATen/native/vulkan/ops/VulkanPackedContext.h lines 12–32

class VulkanPackedContext {
 protected:
  c10::impl::GenericList packed_;

 public:
  VulkanPackedContext() : packed_{c10::AnyType::get()} {}
  VulkanPackedContext(const VulkanPackedContext&) = default;
  VulkanPackedContext(VulkanPackedContext&&) = default;

  inline const c10::IValue get_val(int64_t i) const {
    return packed_.get(i);
  }

  inline void set_val(int64_t i, const c10::IValue& val) const {
    return packed_.set(i, val);
  }

  virtual const c10::impl::GenericList unpack() const = 0;

  virtual ~VulkanPackedContext() = default;
};