aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/detail/ExecutionHelpers.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/detail/ExecutionHelpers.h')
-rw-r--r--arm_compute/graph/detail/ExecutionHelpers.h7
1 files changed, 4 insertions, 3 deletions
diff --git a/arm_compute/graph/detail/ExecutionHelpers.h b/arm_compute/graph/detail/ExecutionHelpers.h
index 3a357776e4..fd8d082770 100644
--- a/arm_compute/graph/detail/ExecutionHelpers.h
+++ b/arm_compute/graph/detail/ExecutionHelpers.h
@@ -71,12 +71,13 @@ void allocate_const_tensors(Graph &g);
void allocate_all_tensors(Graph &g);
/** Configures all nodes of graph
*
- * @param[in] g Graph to configure the nodes
- * @param[in] ctx Graph context to use
+ * @param[in, out] g Graph to configure the nodes
+ * @param[in] ctx Graph context to use
+ * @param[in] node_order The order to configure the nodes
*
* @return The execution workload
*/
-ExecutionWorkload configure_all_nodes(Graph &g, GraphContext &ctx);
+ExecutionWorkload configure_all_nodes(Graph &g, GraphContext &ctx, const std::vector<NodeID> &node_order);
/** Release the memory of all unused const nodes
*
* @param[in] g Graph to release the memory from