24 #ifndef __ARM_COMPUTE_TENSORALLOCATOR_H__ 25 #define __ARM_COMPUTE_TENSORALLOCATOR_H__ 60 uint8_t *
data()
const;
74 uint8_t *lock()
override;
77 void unlock()
override;
80 std::shared_ptr<std::vector<uint8_t>> _buffer;
TensorAllocator()
Default constructor.
Basic implementation of a CPU memory tensor allocator.
void allocate() override
Allocate size specified by TensorInfo of CPU memory.
uint8_t * data() const
Returns the pointer to the allocated data.
void init(const TensorInfo &input)
Initialize a tensor based on the passed TensorInfo.
void init(const TensorAllocator &allocator, const Coordinates &coords, TensorInfo sub_info)
Shares the same backing memory with another tensor allocator, while the tensor info might be differen...
Store the tensor's metadata.
Interface to allocate tensors.