diff options
author | Дилшоджон Умронхонович Пошшоев/AI Tools Lab /SRR/Engineer/삼성전자 <d.poshshoev@samsung.com> | 2019-03-25 12:33:57 +0300 |
---|---|---|
committer | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-03-25 18:33:57 +0900 |
commit | b06ae1990e46c91b9698bc43c8b828b7681256c3 (patch) | |
tree | c2c9e34893c2926fdb596087e4f3566722f5ce2e /runtimes | |
parent | 0da491211e369f0e4b4f37960c1da0fbd0e51493 (diff) | |
download | nnfw-b06ae1990e46c91b9698bc43c8b828b7681256c3.tar.gz nnfw-b06ae1990e46c91b9698bc43c8b828b7681256c3.tar.bz2 nnfw-b06ae1990e46c91b9698bc43c8b828b7681256c3.zip |
Redo TensorBuilderSet as unordered_set (#4836)
Redo TensorBuilderSet as unordered_set since we don't care if it is
sorted or not
Signed-off-by: Poshshoev Dilshodzhon <d.poshshoev@samsung.com>
Diffstat (limited to 'runtimes')
-rw-r--r-- | runtimes/neurun/core/include/backend/ITensorBuilder.h | 4 | ||||
-rw-r--r-- | runtimes/neurun/core/src/compiler/Compiler.cc | 2 | ||||
-rw-r--r-- | runtimes/neurun/core/src/exec/DataflowExecutor.h | 1 |
3 files changed, 3 insertions, 4 deletions
diff --git a/runtimes/neurun/core/include/backend/ITensorBuilder.h b/runtimes/neurun/core/include/backend/ITensorBuilder.h index 2f7165f45..514ec3a18 100644 --- a/runtimes/neurun/core/include/backend/ITensorBuilder.h +++ b/runtimes/neurun/core/include/backend/ITensorBuilder.h @@ -63,7 +63,7 @@ struct ITensorBuilder } // namespace backend } // namespace neurun -#include <set> +#include <unordered_set> #include <memory> namespace neurun @@ -71,7 +71,7 @@ namespace neurun namespace backend { -using TensorBuilderSet = std::set<std::shared_ptr<backend::ITensorBuilder>>; +using TensorBuilderSet = std::unordered_set<std::shared_ptr<backend::ITensorBuilder>>; } // namespace backend } // namespace neurun diff --git a/runtimes/neurun/core/src/compiler/Compiler.cc b/runtimes/neurun/core/src/compiler/Compiler.cc index edef0d0c4..270f562da 100644 --- a/runtimes/neurun/core/src/compiler/Compiler.cc +++ b/runtimes/neurun/core/src/compiler/Compiler.cc @@ -159,7 +159,7 @@ std::shared_ptr<exec::IExecutor> Compiler::createDataflowExecutor(graph::Graph & stages[index] = stage_gen->generate(node); }); - std::unordered_set<std::shared_ptr<backend::ITensorBuilder>> tensor_builders; + backend::TensorBuilderSet tensor_builders; model.operands().iterate( [&](const model::operand::Index &ind, const model::operand::Object &obj) { diff --git a/runtimes/neurun/core/src/exec/DataflowExecutor.h b/runtimes/neurun/core/src/exec/DataflowExecutor.h index 367716227..f9a76f332 100644 --- a/runtimes/neurun/core/src/exec/DataflowExecutor.h +++ b/runtimes/neurun/core/src/exec/DataflowExecutor.h @@ -19,7 +19,6 @@ #include <list> #include <queue> -#include <unordered_set> #include <unordered_map> #include "FunctionSequence.h" |