[nnc] Fix for constant in ModelAnalyzer (#2744)
authorАндрей Шедько/AI Tools Lab /SRR/Engineer/삼성전자 <a.shedko@samsung.com>
Wed, 9 Jan 2019 08:29:33 +0000 (11:29 +0300)
committerРоман Михайлович Русяев/AI Tools Lab /SRR/Staff Engineer/삼성전자 <r.rusyaev@samsung.com>
Wed, 9 Jan 2019 08:29:33 +0000 (11:29 +0300)
Fixed constants not being visible in soft backend

Signed-off-by: Andrei Shedko <a.shedko@samsung.com>
contrib/nnc/passes/soft_backend/ModelAnalyzer.cpp

index bd0c827..b4fb578 100644 (file)
@@ -140,6 +140,8 @@ void ModelAnalyzer::analyze(const mir::Graph* g) {
 
   // Collect all inputs and constants
   vector<Operation*> init_ops(g->collectInputs());
+  auto constants = g->collectConstants();
+  init_ops.insert(init_ops.end(), constants.begin(), constants.end());
 
   // Walk all network inputs
   for (Operation* in : init_ops) {