diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2020-08-14 15:19:19 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2020-08-14 15:19:19 +0900 |
commit | 042b262b3633b6c0f577aed6cb4b980ad0c1dcf3 (patch) | |
tree | e79fb9ffe65b21bdc5863306db2757ab187a3306 /runtime/onert/backend/cpu/TensorBuilder.cc | |
parent | 05e0ec30a632339a8533082476f27bda31ccde16 (diff) | |
download | nnfw-042b262b3633b6c0f577aed6cb4b980ad0c1dcf3.tar.gz nnfw-042b262b3633b6c0f577aed6cb4b980ad0c1dcf3.tar.bz2 nnfw-042b262b3633b6c0f577aed6cb4b980ad0c1dcf3.zip |
Imported Upstream version 1.8.0upstream/1.8.0submit/tizen/20200814.062151
Diffstat (limited to 'runtime/onert/backend/cpu/TensorBuilder.cc')
-rw-r--r-- | runtime/onert/backend/cpu/TensorBuilder.cc | 20 |
1 files changed, 8 insertions, 12 deletions
diff --git a/runtime/onert/backend/cpu/TensorBuilder.cc b/runtime/onert/backend/cpu/TensorBuilder.cc index 886e8d820..ab8ba5756 100644 --- a/runtime/onert/backend/cpu/TensorBuilder.cc +++ b/runtime/onert/backend/cpu/TensorBuilder.cc @@ -29,8 +29,8 @@ namespace cpu TensorBuilder::TensorBuilder() : _tensor_reg{new cpu_common::TensorRegistry()}, - _static_tensor_mgr{new cpu_common::StaticTensorManager(_tensor_reg)}, - _dynamic_tensor_mgr{new cpu_common::DynamicTensorManager(_tensor_reg)} + _dynamic_tensor_mgr{new cpu_common::DynamicTensorManager(_tensor_reg)}, + _static_tensor_mgr{new StaticTensorManager(_tensor_reg, _dynamic_tensor_mgr.get())} { /* empty */ } @@ -77,11 +77,7 @@ bool TensorBuilder::isRegistered(const ir::OperandIndex &ind) const return _tensor_info_map.find(ind) != _tensor_info_map.end(); } -void TensorBuilder::prepare(void) -{ - _static_tensor_mgr->allocateConsts(); - _static_tensor_mgr->allocateNonconsts(); -} +void TensorBuilder::prepare(void) { _static_tensor_mgr->allocateNonconsts(); } void TensorBuilder::allocate() { @@ -99,17 +95,17 @@ std::shared_ptr<IPortableTensor> TensorBuilder::portableAt(const ir::OperandInde return _tensor_reg->getPortableTensor(ind); } -bool TensorBuilder::setExternalTensor(const ir::OperandIndex &ind, - const std::shared_ptr<IPortableTensor> &tensor) +bool TensorBuilder::setMigrantTensor(const ir::OperandIndex &ind, + const std::shared_ptr<IPortableTensor> &tensor) { - return _tensor_reg->setExternalTensor(ind, tensor); + return _tensor_reg->setMigrantTensor(ind, tensor); } void TensorBuilder::iterate(const IterateFunction &fn) { _static_tensor_mgr->iterate(fn); } -std::shared_ptr<cpu_common::Tensor> TensorBuilder::at(const ir::OperandIndex &ind) +std::shared_ptr<Tensor> TensorBuilder::at(const ir::OperandIndex &ind) { - return _tensor_reg->getManagedTensor(ind); + return _tensor_reg->getNativeTensor(ind); } std::unique_ptr<ITensorManager> TensorBuilder::releaseStaticTensorManager(void) |