diff options
Diffstat (limited to 'runtime/onert/core/src/backend/basic/TensorBuilder.cc')
-rw-r--r-- | runtime/onert/core/src/backend/basic/TensorBuilder.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/runtime/onert/core/src/backend/basic/TensorBuilder.cc b/runtime/onert/core/src/backend/basic/TensorBuilder.cc index a10cc2bf9..f9d83875d 100644 --- a/runtime/onert/core/src/backend/basic/TensorBuilder.cc +++ b/runtime/onert/core/src/backend/basic/TensorBuilder.cc @@ -34,6 +34,14 @@ TensorBuilder::TensorBuilder(const std::shared_ptr<TensorRegistry> &tensor_reg) /* empty */ } +TensorBuilder::TensorBuilder(const std::shared_ptr<TensorRegistry> &tensor_reg, + const std::string planner_id) + : _tensor_reg{tensor_reg}, _dynamic_tensor_mgr{new DynamicTensorManager(_tensor_reg)}, + _static_tensor_mgr{new StaticTensorManager(_tensor_reg, planner_id, _dynamic_tensor_mgr.get())} +{ + /* empty */ +} + void TensorBuilder::registerTensorInfo(const ir::OperandIndex &ind, const ir::OperandInfo &info, ir::Layout layout) { |