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

1 min read time

Applies to Linux

• include
• core

tensor_buffer.hpp

Go to the documentation of this file.

1 

23 #pragma once

24 

25 #include <stdint.h>

26 

27 #include <array>

28 

29 #include "tensor_layout.hpp"

30 

31 namespace roccv {

32 

38 struct TensorBuffer {

39  void* basePtr;

40 };

41 

46 struct TensorBufferStrided : public TensorBuffer {

47  std::array<int64_t, ROCCV_TENSOR_MAX_RANK> strides;

48 };

49 

50 } // namespace roccv

roccv

Definition: strided_data_wrap.hpp:33

roccv::TensorBuffer

A tensor buffer. Stores the raw data associated with a tensor. Data should be allocated by the user a...

Definition: tensor_buffer.hpp:38

roccv::TensorBuffer::basePtr

void * basePtr

Definition: tensor_buffer.hpp:39

roccv::TensorBufferStrided

A tensor buffer with strided data.

Definition: tensor_buffer.hpp:46

roccv::TensorBufferStrided::strides

std::array< int64_t, ROCCV_TENSOR_MAX_RANK > strides

Definition: tensor_buffer.hpp:47

tensor_layout.hpp