From e043767d068da389308507011d944e6db9e4d676 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 2 May 2018 14:07:55 +0100 Subject: COMPMID-920: Introduce prepare() stage Change-Id: I08ddb7f6e061178e7566518b48e4e18f8f078596 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/129825 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- arm_compute/graph/detail/ExecutionHelpers.h | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'arm_compute/graph/detail/ExecutionHelpers.h') diff --git a/arm_compute/graph/detail/ExecutionHelpers.h b/arm_compute/graph/detail/ExecutionHelpers.h index 52304d6836..a868df8a5d 100644 --- a/arm_compute/graph/detail/ExecutionHelpers.h +++ b/arm_compute/graph/detail/ExecutionHelpers.h @@ -35,6 +35,7 @@ class Graph; class GraphContext; class ExecutionWorkload; class Tensor; +class INode; namespace detail { @@ -45,6 +46,21 @@ void default_initialize_backends(); * @param[in] g Graph to configure */ void configure_all_tensors(Graph &g); +/** Allocates all input tensors of a node. + * + * @param[in] node Node to allocate the input tensor of + */ +void allocate_all_input_tensors(INode &node); +/** Allocates all output tensors of a node. + * + * @param[in] node Node to allocate the output tensor of + */ +void allocate_all_output_tensors(INode &node); +/** Allocates const tensor of a given graph + * + * @param[in] g Graph to allocate the tensors + */ +void allocate_const_tensors(Graph &g); /** Allocates all tensors of a graph * * @param[in] g Graph to allocate the tensors @@ -88,6 +104,11 @@ void call_all_input_node_accessors(ExecutionWorkload &workload); * @param[in] workload Workload to execute */ void call_all_output_node_accessors(ExecutionWorkload &workload); +/** Prepares all tasks for execution + * + * @param[in] workload Workload to prepare + */ +void prepare_all_tasks(ExecutionWorkload &workload); /** Executes all tasks of a workload * * @param[in] workload Workload to execute -- cgit v1.2.1