diff options
Diffstat (limited to 'src/core/ITensorPack.cpp')
| -rw-r--r-- | src/core/ITensorPack.cpp | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/core/ITensorPack.cpp b/src/core/ITensorPack.cpp index 546f669985..9eaeece271 100644 --- a/src/core/ITensorPack.cpp +++ b/src/core/ITensorPack.cpp @@ -27,14 +27,23 @@ namespace arm_compute { +ITensorPack::ITensorPack(std::initializer_list<PackElement> l) + : _pack() +{ + for(auto &e : l) + { + _pack[e.id] = e; + } +} + void ITensorPack::add_tensor(int id, ITensor *tensor) { - _pack[id] = PackElement(tensor); + _pack[id] = PackElement(id, tensor); } void ITensorPack::add_tensor(int id, const ITensor *tensor) { - _pack[id] = PackElement(tensor); + _pack[id] = PackElement(id, tensor); } void ITensorPack::add_const_tensor(int id, const ITensor *tensor) |