rocalTensor Class Reference#
rocalTensor Class Referenceabstract
class representing rocal tensor More...
#include <rocal_api_tensor.h>
Public Member Functions | |
| virtual void * | buffer ()=0 |
| virtual unsigned | copy_data (void *user_buffer, RocalOutputMemType external_mem_type=ROCAL_MEMCPY_HOST)=0 |
| virtual unsigned | copy_data (void *user_buffer, uint x_offset, uint y_offset, uint max_cols, uint max_rows)=0 |
| virtual unsigned | num_of_dims ()=0 |
| virtual unsigned | batch_size ()=0 |
| virtual std::vector< size_t > | dims ()=0 |
| virtual std::vector< size_t > | strides ()=0 |
| virtual RocalTensorLayout | layout ()=0 |
| virtual RocalTensorBackend | backend ()=0 |
| virtual RocalTensorOutputType | data_type ()=0 |
| virtual RocalOutputMemType | mem_type ()=0 |
| virtual size_t | data_size ()=0 |
| virtual RocalROICordsType | roi_type ()=0 |
| virtual size_t | get_roi_dims_size ()=0 |
| virtual void | copy_roi (void *roi_buffer)=0 |
| virtual std::vector< size_t > | shape ()=0 |
| virtual void | set_dims (std::vector< size_t > dims)=0 |
| virtual void | set_mem_handle (void *buffer)=0 |
| virtual void | set_tensor_layout (RocalTensorLayout layout)=0 |
| virtual uint64_t | data_type_size ()=0 |
Detailed Description
class representing rocal tensor
The documentation for this class was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/develop/rocAL/include/api/rocal_api_tensor.h