/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_data.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_data.hpp Source File#

4 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_data.hpp Source File

Go to the documentation of this file.

  
 #pragma once
  
 #include <stdint.h>
  
 #include <iostream>
 #include <optional>
  
 #include "data_type.hpp"
 #include "tensor_buffer.hpp"
 #include "tensor_shape.hpp"
  
 namespace roccv {
  
 class TensorData {
    public:
     TensorData() = delete;
     virtual ~TensorData() = default;
  
     virtual int rank() const;
  
     virtual const TensorShape &shape() const &;
  
     virtual const int64_t shape(int d) const &;
  
     virtual const DataType &dtype() const;
  
     virtual void *basePtr() const;
  
     virtual const eDeviceType device() const;
  
     template <typename Derived>
     std::optional<Derived> cast() {
         static_assert(std::is_base_of<TensorData, Derived>::value, "Cannot cast TensorData to an unrelated type.");
         static_assert(sizeof(Derived) == sizeof(TensorData), "Derived type must not add any additional data members.");
         return std::optional(Derived(m_shape, m_dtype, m_buffer, m_deviceType));
     }
  
    protected:
     TensorData(const TensorShape &tshape, const DataType &dtype, const TensorBufferStrided &buffer,
                const eDeviceType device = eDeviceType::GPU);
  
     TensorShape m_shape;
     DataType m_dtype;
     eDeviceType m_deviceType;
     TensorBufferStrided m_buffer;
 };
  
 class TensorDataStrided : public TensorData {
    public:
     using Buffer = TensorBufferStrided;
     TensorDataStrided(const TensorShape &tshape, const DataType &dtype, const TensorBufferStrided &buffer,
                       const eDeviceType device = eDeviceType::GPU);
  
     const int64_t stride(int d) const;
 };
 }  // namespace roccv