[neurun] Remove InitializerGenerator dependency (#3508)

Remove all InitializerGenerator dependent code since we now have
ConstantInitializer for this purpose.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/runtimes/neurun/src/backend/BackendManager.cc b/runtimes/neurun/src/backend/BackendManager.cc
index 8fef674..0dbb14c 100644 (file)
--- a/runtimes/neurun/src/backend/BackendManager.cc
+++ b/runtimes/neurun/src/backend/BackendManager.cc
@@ -18,11 +18,9 @@
 
 #include "backend/acl_cl/Config.h"
 #include "backend/acl_cl/TensorBuilder.h"
-#include "backend/acl_cl/InitializerGenerator.h"
 #include "backend/acl_cl/StageGenerator.h"
 #include "backend/cpu/Config.h"
 #include "backend/cpu/TensorBuilder.h"
-#include "backend/cpu/InitializerGenerator.h"
 #include "backend/cpu/StageGenerator.h"
 
 namespace neurun
@@ -31,20 +29,14 @@ namespace backend
 {
 
 Backend::Backend(const std::shared_ptr<neurun::backend::IConfig> &backend_config,
-                 const std::shared_ptr<neurun::backend::IInitializerGenerator> &initializer_gen,
                  const std::shared_ptr<neurun::backend::IStageGenerator> &stage_gen)
-    : _config(backend_config), _initializer_gen(initializer_gen), _stage_gen(stage_gen)
+    : _config(backend_config), _stage_gen(stage_gen)
 {
   backend_config->initialize();
 }
 
 const std::shared_ptr<neurun::backend::IConfig> Backend::config() const { return _config; }
 
-const std::shared_ptr<neurun::backend::IInitializerGenerator> Backend::initializer_gen() const
-{
-  return _initializer_gen;
-}
-
 const std::shared_ptr<neurun::backend::IStageGenerator> Backend::stage_gen() const
 {
   return _stage_gen;
@@ -62,10 +54,9 @@ BackendManager::BackendManager(const neurun::graph::operand::Set &operands)
     using namespace ::neurun::backend::acl_cl;
     auto config = std::make_shared<Config>();
     auto tensor_builder = std::make_shared<TensorBuilder>();
-    auto initializer_gen = std::make_shared<InitializerGenerator>(operands);
     auto stage_gen = std::make_shared<StageGenerator>(operands, tensor_builder);
 
-    _gen_map[config->id()] = {config, initializer_gen, stage_gen};
+    _gen_map[config->id()] = {config, stage_gen};
   }
 
   // Add CPU backend
@@ -73,10 +64,9 @@ BackendManager::BackendManager(const neurun::graph::operand::Set &operands)
     using namespace ::neurun::backend::cpu;
     auto config = std::make_shared<Config>();
     auto tensor_builder = std::make_shared<TensorBuilder>();
-    auto initializer_gen = std::make_shared<InitializerGenerator>(operands);
     auto stage_gen = std::make_shared<StageGenerator>(operands, tensor_builder);
 
-    _gen_map[config->id()] = {config, initializer_gen, stage_gen};
+    _gen_map[config->id()] = {config, stage_gen};
   }
 }
 

diff --git a/runtimes/neurun/src/backend/BackendManager.h b/runtimes/neurun/src/backend/BackendManager.h
index 50cece9..aeb4153 100644 (file)
--- a/runtimes/neurun/src/backend/BackendManager.h
+++ b/runtimes/neurun/src/backend/BackendManager.h
@@ -28,7 +28,6 @@ namespace backend
 {
 
 struct IConfig;
-struct IInitializerGenerator;
 struct IStageGenerator;
 struct ITensorBuilder;
 
@@ -36,23 +35,20 @@ class Backend
 {
 public:
   Backend(const std::shared_ptr<neurun::backend::IConfig> &backend_config,
-          const std::shared_ptr<neurun::backend::IInitializerGenerator> &initializer_gen,
           const std::shared_ptr<neurun::backend::IStageGenerator> &stage_gen);
 
-  Backend(void) : _config(nullptr), _initializer_gen(nullptr), _stage_gen(nullptr)
+  Backend(void) : _config(nullptr), _stage_gen(nullptr)
   {
     // DO NOTHING
   }
 
 public:
   const std::shared_ptr<neurun::backend::IConfig> config() const;
-  const std::shared_ptr<neurun::backend::IInitializerGenerator> initializer_gen() const;
   const std::shared_ptr<neurun::backend::IStageGenerator> stage_gen() const;
   const std::shared_ptr<neurun::backend::ITensorBuilder> tensor_builder() const;
 
 private:
   std::shared_ptr<neurun::backend::IConfig> _config;
-  std::shared_ptr<neurun::backend::IInitializerGenerator> _initializer_gen;
   std::shared_ptr<neurun::backend::IStageGenerator> _stage_gen;
 };
 

diff --git a/runtimes/neurun/src/codegen/BackendResolver.h b/runtimes/neurun/src/codegen/BackendResolver.h
index e788836..4ec7c52 100644 (file)
--- a/runtimes/neurun/src/codegen/BackendResolver.h
+++ b/runtimes/neurun/src/codegen/BackendResolver.h
@@ -24,7 +24,6 @@
 #include "logging.h"
 #include "util/EnvVar.h"
 #include "backend/BackendManager.h"
-#include "backend/interface/IInitializerGenerator.h"
 #include "backend/interface/IStageGenerator.h"
 
 namespace neurun

diff --git a/runtimes/neurun/src/codegen/Planner.cc b/runtimes/neurun/src/codegen/Planner.cc
index 9393156..addab58 100644 (file)
--- a/runtimes/neurun/src/codegen/Planner.cc
+++ b/runtimes/neurun/src/codegen/Planner.cc
@@ -35,10 +35,6 @@ void Planner::visit(const graph::operation::Conv2DNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -49,10 +45,6 @@ void Planner::visit(const graph::operation::MaxPool2DNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -63,10 +55,6 @@ void Planner::visit(const graph::operation::AvgPool2DNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -82,10 +70,6 @@ void Planner::visit(const graph::operation::ConcatNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -98,10 +82,6 @@ void Planner::visit(const graph::operation::FullyConnectedNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -131,10 +111,6 @@ void Planner::visit(const graph::operation::ReshapeNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -158,10 +134,6 @@ void Planner::visit(const graph::operation::SoftmaxNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -172,10 +144,6 @@ void Planner::visit(const graph::operation::NOPNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -193,10 +161,6 @@ void Planner::visit(const graph::operation::PermuteNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));
@@ -209,10 +173,6 @@ void Planner::visit(const graph::operation::AddNode &node)
   // backend
   auto backend = node.lower_info()->backend();
 
-  // Generate Initializers
-  auto init_gen = backend->initializer_gen();
-  _builder.addInitializer(init_gen->generate(node));
-
   // Generate Stage
   auto stage_gen = backend->stage_gen();
   _builder.addStage(stage_gen->generate(node));