diff options
Diffstat (limited to 'arm_compute/graph/Graph.h')
| -rw-r--r-- | arm_compute/graph/Graph.h | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/arm_compute/graph/Graph.h b/arm_compute/graph/Graph.h index d8d3feb1f7..e6e173f5fa 100644 --- a/arm_compute/graph/Graph.h +++ b/arm_compute/graph/Graph.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 Arm Limited. + * Copyright (c) 2018-2020,2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -30,7 +30,6 @@ #include "arm_compute/graph/Types.h" #include "support/Mutex.h" -#include "support/ToolchainSupport.h" #include <map> #include <memory> @@ -80,7 +79,7 @@ public: * @return ID of the node */ template <typename NT, typename... Ts> - NodeID add_node(Ts &&... args); + NodeID add_node(Ts &&...args); /** Remove the node with the given ID * * @param[in] nid ID of the node to remove @@ -222,17 +221,17 @@ private: TensorID create_tensor(const TensorDescriptor &desc = TensorDescriptor()); private: - GraphID _id = GraphID(0); /**< Graph id */ - std::string _name = {}; /**< Graph name */ - std::vector<std::unique_ptr<INode>> _nodes = {}; /**< Graph nodes */ - std::vector<std::unique_ptr<Edge>> _edges = {}; /**< Graph edges */ - std::vector<std::unique_ptr<Tensor>> _tensors = {}; /**< Graph tensors */ + GraphID _id = GraphID(0); /**< Graph id */ + std::string _name = {}; /**< Graph name */ + std::vector<std::unique_ptr<INode>> _nodes = {}; /**< Graph nodes */ + std::vector<std::unique_ptr<Edge>> _edges = {}; /**< Graph edges */ + std::vector<std::unique_ptr<Tensor>> _tensors = {}; /**< Graph tensors */ std::map<NodeType, std::vector<NodeID>> _tagged_nodes = {}; /**< Graph nodes map with the node type as key */ - arm_compute::Mutex _mtx = {}; /**< Mutex used for graph construction */ + arm_compute::Mutex _mtx = {}; /**< Mutex used for graph construction */ }; template <typename NT, typename... Ts> -inline NodeID Graph::add_node(Ts &&... args) +inline NodeID Graph::add_node(Ts &&...args) { arm_compute::lock_guard<arm_compute::Mutex> lock(_mtx); @@ -246,7 +245,7 @@ inline NodeID Graph::add_node(Ts &&... args) _tagged_nodes[node->type()].push_back(nid); // Associate a new tensor with each output - for(auto &output : node->_outputs) + for (auto &output : node->_outputs) { output = create_tensor(); } |