/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_storage.hpp Source File#
3 min read time
Applies to Linux
tensor_storage.hpp
Go to the documentation of this file.
44 explicit TensorStorage(void* data, eDeviceType device, eOwnership ownership = eOwnership::OWNING);
Stores the underlying data of a tensor and is responsible for allocation/freeing of tensor memory....
Definition: tensor_storage.hpp:34
TensorStorage(void *data, eDeviceType device, eOwnership ownership=eOwnership::OWNING)
Creates a new TensorStorage object and takes ownership of the data pointer.
TensorStorage(size_t bytes, eDeviceType device)
Creates a new TensorStorage object and allocates the requested number of bytes.
~TensorStorage()
const eDeviceType device() const
Retrieves the device that the tensor data is allocated on.
Definition: strided_data_wrap.hpp:33
eOwnership
Describes whether a container should own a resource or only be a view for a resource.
Definition: util_enums.h:75
@ OWNING
eDeviceType
Describes the device type. Used to determine where Tensor data should be allocated and whether operat...
Definition: util_enums.h:69