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

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

2 min read time

Applies to Linux

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

Go to the documentation of this file.

  
 #pragma once
  
 #include <span>
  
 #include "tensor_layout.hpp"
  
 namespace roccv {
 class TensorShape {
    public:
     TensorShape() = delete;
  
     TensorShape(const TensorLayout &layout,
                 const std::span<const int64_t> shape);
     TensorShape(const TensorLayout &layout,
                 const std::initializer_list<const int64_t> shape);
  
     const TensorLayout &layout() const;
  
     int size() const;
  
     // Operators
     int64_t operator[](int32_t i) const;
     TensorShape &operator=(const TensorShape &other);
     bool operator==(const TensorShape &rhs) const;
     bool operator!=(const TensorShape &rhs) const;
  
    private:
     std::array<int64_t, ROCCV_TENSOR_MAX_RANK> m_shape;
     TensorLayout m_layout;
     size_t m_size;
 };
 }  // namespace roccv