From: 이한종/동작제어Lab(SR)/Engineer/삼성전자 Date: Thu, 6 Sep 2018 01:33:22 +0000 (+0900) Subject: [neurun] Add IBackendConfig getter to Backend (#2611) X-Git-Tag: 0.2~72 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=75a4dd527c181250b1cd2dab260874c153be4678;p=platform%2Fcore%2Fml%2Fnnfw.git [neurun] Add IBackendConfig getter to Backend (#2611) Add `IBackendConfig` getter `config()` method to `Backend`. We need to keep it otherwise we do not have any way to call `getOperandLayout()`. Signed-off-by: Hanjoung Lee --- diff --git a/runtimes/neurun/src/backend/BackendManager.cc b/runtimes/neurun/src/backend/BackendManager.cc index 46adee5..d5ae3eb 100644 --- a/runtimes/neurun/src/backend/BackendManager.cc +++ b/runtimes/neurun/src/backend/BackendManager.cc @@ -17,11 +17,13 @@ namespace backend Backend::Backend(const std::shared_ptr &backend_config, const std::shared_ptr &initializer_gen, const std::shared_ptr &stage_gen) - : _initializer_gen(initializer_gen), _stage_gen(stage_gen) + : _config(backend_config), _initializer_gen(initializer_gen), _stage_gen(stage_gen) { backend_config->initialize(); } +const std::shared_ptr Backend::config() const { return _config; } + const std::shared_ptr Backend::initializer_gen() const { return _initializer_gen; diff --git a/runtimes/neurun/src/backend/BackendManager.h b/runtimes/neurun/src/backend/BackendManager.h index ca047d8..21c98a8 100644 --- a/runtimes/neurun/src/backend/BackendManager.h +++ b/runtimes/neurun/src/backend/BackendManager.h @@ -23,17 +23,19 @@ public: const std::shared_ptr &initializer_gen, const std::shared_ptr &stage_gen); - Backend(void) : _initializer_gen(nullptr), _stage_gen(nullptr) + Backend(void) : _config(nullptr), _initializer_gen(nullptr), _stage_gen(nullptr) { // DO NOTHING } public: + const std::shared_ptr config() const; const std::shared_ptr initializer_gen() const; const std::shared_ptr stage_gen() const; const std::shared_ptr tensor_builder() const; private: + std::shared_ptr _config; std::shared_ptr _initializer_gen; std::shared_ptr _stage_gen; };