aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/nodes/Nodes.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/nodes/Nodes.h')
-rw-r--r--arm_compute/graph/nodes/Nodes.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/graph/nodes/Nodes.h b/arm_compute/graph/nodes/Nodes.h
index 59ac131b8c..92dfa41fc3 100644
--- a/arm_compute/graph/nodes/Nodes.h
+++ b/arm_compute/graph/nodes/Nodes.h
@@ -47,5 +47,6 @@
#include "arm_compute/graph/nodes/ResizeLayerNode.h"
#include "arm_compute/graph/nodes/SoftmaxLayerNode.h"
#include "arm_compute/graph/nodes/SplitLayerNode.h"
+#include "arm_compute/graph/nodes/YOLOLayerNode.h"
#endif /* __ARM_COMPUTE_GRAPH_NODES_H__ */