diff options
Diffstat (limited to 'arm_compute/graph/Workload.h')
-rw-r--r-- | arm_compute/graph/Workload.h | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/arm_compute/graph/Workload.h b/arm_compute/graph/Workload.h index b19c932636..11bb22ea9a 100644 --- a/arm_compute/graph/Workload.h +++ b/arm_compute/graph/Workload.h @@ -37,6 +37,7 @@ namespace graph class ITensorHandle; class INode; class Tensor; +class Graph; /** Execution task * @@ -52,14 +53,18 @@ struct ExecutionTask /** Function operator */ void operator()(); + + /** Prepare execution task */ + void prepare(); }; /** Execution workload */ struct ExecutionWorkload { - std::vector<Tensor *> inputs = {}; /**< Input handles */ - std::vector<Tensor *> outputs = {}; /**< Output handles */ - std::vector<ExecutionTask> tasks = {}; /**< Execution workload */ + std::vector<Tensor *> inputs = {}; /**< Input handles */ + std::vector<Tensor *> outputs = {}; /**< Output handles */ + std::vector<ExecutionTask> tasks = {}; /**< Execution workload */ + Graph *graph = nullptr; /**< Graph bound to the workload */ }; } // namespace graph } // namespace arm_compute |