GenericTensorWrapper< T > Class Template Reference

2 min read time

Applies to Linux

• roccv
• GenericTensorWrapper

Public Member Functions | Public Attributes | List of all members

roccv::GenericTensorWrapper< T > Class Template Reference

#include <generic_tensor_wrapper.hpp>

Public Member Functions
template<typename... ARGS>
__device__ __host__ T &	at (ARGS... idx)
	GenericTensorWrapper (const Tensor &tensor)
	Constructs a generic tensor descriptor from a roccv::Tensor. More...

Public Attributes
unsigned char *	data
int	rank
std::array< int64_t, ROCCV_TENSOR_MAX_RANK >	shape
std::array< int64_t, ROCCV_TENSOR_MAX_RANK >	strides

Constructor & Destructor Documentation

◆ GenericTensorWrapper()

template<typename T >

roccv::GenericTensorWrapper< T >::GenericTensorWrapper ( const Tensor &  tensor )

inline

Constructs a generic tensor descriptor from a roccv::Tensor.

Parameters

tensor

A roccv::Tensor to wrap.

Member Function Documentation

◆ at()

template<typename T >

template<typename... ARGS>

__device__ __host__ T& roccv::GenericTensorWrapper< T >::at ( ARGS...  idx )

inline

Member Data Documentation

◆ data

template<typename T >

unsigned char* roccv::GenericTensorWrapper< T >::data

◆ rank

template<typename T >

int roccv::GenericTensorWrapper< T >::rank

◆ shape

template<typename T >

std::array<int64_t, ROCCV_TENSOR_MAX_RANK> roccv::GenericTensorWrapper< T >::shape

◆ strides

template<typename T >

std::array<int64_t, ROCCV_TENSOR_MAX_RANK> roccv::GenericTensorWrapper< T >::strides

---

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

• /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/wrappers/generic_tensor_wrapper.hpp