AI Engine API User Guide (AIE) 2023.2
Loading...
Searching...
No Matches
aie::tensor_descriptor< Rank, T, Elems, NativeRepr > Class Template Reference

#include <aie.hpp>

Public Types

using type = T
 
using vector_type = vector< T, Elems >
 

Public Member Functions

constexpr tensor_descriptor (const std::array< tensor_dim, Rank > &dims, bool byte_steps=false)
 
constexpr tensor_descriptor (const tensor_descriptor &other)
 

Static Public Attributes

static constexpr unsigned elems = Elems
 

Friends

template<aie_dm_resource Resource, typename T2 , typename TensorDescriptor >
constexpr auto make_restrict_tensor_buffer_stream (T2 *base, const TensorDescriptor &dims)
 
template<aie_dm_resource Resource, typename T2 , typename TensorDescriptor >
constexpr auto make_tensor_buffer_stream (T2 *base, const TensorDescriptor &dims)
 
template<typename T2 , unsigned Elems2, typename... Args>
constexpr auto make_tensor_descriptor_from_native (Args &&... args)
 
template<typename T2 , unsigned Elems2, typename... Args>
constexpr auto make_tensor_descriptor_from_native_bytes (Args &&... args)
 

Member Typedef Documentation

◆ type

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
using aie::tensor_descriptor< Rank, T, Elems, NativeRepr >::type = T

◆ vector_type

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
using aie::tensor_descriptor< Rank, T, Elems, NativeRepr >::vector_type = vector<T, Elems>

Constructor & Destructor Documentation

◆ tensor_descriptor() [1/2]

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
constexpr aie::tensor_descriptor< Rank, T, Elems, NativeRepr >::tensor_descriptor ( const std::array< tensor_dim, Rank > &  dims,
bool  byte_steps = false 
)
inlineexplicitconstexpr

◆ tensor_descriptor() [2/2]

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
constexpr aie::tensor_descriptor< Rank, T, Elems, NativeRepr >::tensor_descriptor ( const tensor_descriptor< Rank, T, Elems, NativeRepr > &  other)
inlineconstexpr

Friends And Related Symbol Documentation

◆ make_restrict_tensor_buffer_stream

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
template<aie_dm_resource Resource, typename T2 , typename TensorDescriptor >
constexpr auto make_restrict_tensor_buffer_stream ( T2 *  base,
const TensorDescriptor &  dims 
)
friend

◆ make_tensor_buffer_stream

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
template<aie_dm_resource Resource, typename T2 , typename TensorDescriptor >
constexpr auto make_tensor_buffer_stream ( T2 *  base,
const TensorDescriptor &  dims 
)
friend

◆ make_tensor_descriptor_from_native

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
template<typename T2 , unsigned Elems2, typename... Args>
constexpr auto make_tensor_descriptor_from_native ( Args &&...  args)
friend

◆ make_tensor_descriptor_from_native_bytes

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
template<typename T2 , unsigned Elems2, typename... Args>
constexpr auto make_tensor_descriptor_from_native_bytes ( Args &&...  args)
friend

Member Data Documentation

◆ elems

template<unsigned Rank, typename T , unsigned Elems, typename NativeRepr = detail::default_repr_t<Rank>>
constexpr unsigned aie::tensor_descriptor< Rank, T, Elems, NativeRepr >::elems = Elems
staticconstexpr

The documentation for this class was generated from the following file: