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

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_requirements.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_requirements.hpp Source File

Go to the documentation of this file.

  
 #pragma once
  
 #include "data_type.hpp"
 #include "tensor_layout.hpp"
  
 namespace roccv {
 struct TensorRequirements {
     // The data type of the tensor.
     DataType dtype;
  
     // Location of the tensor's underlying data.
     eDeviceType device;
  
     // The shape of the tensor.
     TensorShape shape;
  
     // Distance in bytes between each element of a given dimension.
     std::array<int64_t, ROCCV_TENSOR_MAX_RANK> strides;
 };
 }  // namespace roccv