aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/Workload.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/Workload.h')
-rw-r--r--arm_compute/graph/Workload.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/graph/Workload.h b/arm_compute/graph/Workload.h
index 3f70181f59..a36c28a181 100644
--- a/arm_compute/graph/Workload.h
+++ b/arm_compute/graph/Workload.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -25,6 +25,7 @@
#define ARM_COMPUTE_GRAPH_WORKLOAD_H
#include "arm_compute/graph/GraphContext.h"
+#include "arm_compute/graph/Tensor.h"
#include "arm_compute/runtime/IFunction.h"
#include "arm_compute/runtime/IMemoryGroup.h"
@@ -39,7 +40,6 @@ namespace graph
// Forward declarations
class ITensorHandle;
class INode;
-class Tensor;
class Graph;
struct ExecutionTask;