32 _pack[id] = PackElement(tensor);
37 _pack[id] = PackElement(tensor);
42 auto it = _pack.find(
id);
45 return it->second.ctensor !=
nullptr ? it->second.ctensor : it->second.tensor;
52 auto it = _pack.find(
id);
53 return it != _pack.end() ? it->second.tensor :
nullptr;
bool empty() const
Checks if pack is empty.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
const ITensor * get_const_tensor(int id) const
Get constant tensor of a given id.
size_t size() const
Pack size accessor.
ITensor * get_tensor(int id)
Get tensor of a given id from the pac.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.