aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/nodes/NodesFwd.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/nodes/NodesFwd.h')
-rw-r--r--arm_compute/graph/nodes/NodesFwd.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/graph/nodes/NodesFwd.h b/arm_compute/graph/nodes/NodesFwd.h
index 92dc1f0ebd..c000834a86 100644
--- a/arm_compute/graph/nodes/NodesFwd.h
+++ b/arm_compute/graph/nodes/NodesFwd.h
@@ -53,6 +53,7 @@ class ReshapeLayerNode;
class ResizeLayerNode;
class SoftmaxLayerNode;
class SplitLayerNode;
+class YOLOLayerNode;
} // namespace graph
} // namespace arm_compute
#endif /* __ARM_COMPUTE_GRAPH_NODES_FWD_H__ */