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

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

5 min read time

Applies to Linux

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

Go to the documentation of this file.

  
 #pragma once
  
 #include <stdint.h>
  
 #include <unordered_map>
  
 #include "exception.hpp"
 #include "util_enums.h"
  
 #define ROCCV_TENSOR_MAX_RANK (15)
  
 namespace roccv {
 struct TensorLayoutDesc {
     int32_t rank;
     int32_t batch_index;
     int32_t width_index;
     int32_t height_index;
     int32_t channel_index;
     int32_t max_features_index;
     int32_t sift_features_index;
     int32_t sift_octave_layer_index;
 };
  
 class TensorLayout {
    public:
     explicit TensorLayout(eTensorLayout layout) {
         if (TensorLayout::layoutDescriptorTable.count(layout) == 0) {
             throw Exception("Invalid TensorLayout type", eStatusType::INVALID_VALUE);
         }
  
         layout_ = layout;
         layout_desc_ = TensorLayout::layoutDescriptorTable.at(layout);
     }
  
     inline static const std::unordered_map<eTensorLayout, TensorLayoutDesc> layoutDescriptorTable = {
         {TENSOR_LAYOUT_HWC, {3, -1, 1, 0, 2, -1, -1, -1}}, {TENSOR_LAYOUT_NC, {2, 0, -1, -1, 1, -1, -1, -1}},
         {TENSOR_LAYOUT_NW, {2, 0, 1, -1, -1, -1, -1, -1}}, {TENSOR_LAYOUT_NHWC, {4, 0, 2, 1, 3, -1, -1, -1}},
         {TENSOR_LAYOUT_NMC, {3, 0, -1, -1, -1, 1, 2, -1}}, {TENSOR_LAYOUT_NMD, {3, 0, -1, -1, -1, 1, 2, -1}},
         {TENSOR_LAYOUT_LNHWC, {5, 1, 3, 2, 4, -1, -1, 0}}, {TENSOR_LAYOUT_NCHW, {4, 0, 3, 2, 1, -1, -1, -1}},
         {TENSOR_LAYOUT_N, {1, 0, -1, -1, -1, -1, -1, -1}}, {TENSOR_LAYOUT_NWC, {3, 0, 1, -1, 2, -1, -1, -1}}};
  
     eTensorLayout elayout() const { return layout_; }
  
     bool operator==(const eTensorLayout &rhs) const { return this->layout_ == rhs; }
  
     bool operator!=(const eTensorLayout &rhs) const { return !operator==(rhs); }
  
     bool operator==(const TensorLayout &rhs) const { return this->layout_ == rhs.layout_; }
  
     bool operator!=(const TensorLayout &rhs) const { return !operator==(rhs); }
  
     int32_t rank() const { return layout_desc_.rank; }
  
     int32_t batch_index() const { return layout_desc_.batch_index; }
  
     int32_t height_index() const { return layout_desc_.height_index; }
  
     int32_t width_index() const { return layout_desc_.width_index; }
  
     int32_t channels_index() const { return layout_desc_.channel_index; }
  
     int32_t max_features_index() const { return layout_desc_.max_features_index; }
  
     int32_t sift_features_index() const { return layout_desc_.sift_features_index; }
  
     int32_t sift_octave_layer_index() const { return layout_desc_.sift_octave_layer_index; }
  
    private:
     eTensorLayout layout_;
     TensorLayoutDesc layout_desc_;
 };
 }  // namespace roccv