aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/nodes/Nodes.h
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2019-03-22 15:25:32 +0000
committerMichele Di Giorgio <michele.digiorgio@arm.com>2019-03-25 17:16:41 +0000
commitec6997563a7cccf58431267cca39435ecd57cd32 (patch)
treeda355630d63858e31bf84acd5a0f588a1ea4f61f /arm_compute/graph/nodes/Nodes.h
parent2761c2f0b60175469e959982a25ff0abdca6c9ce (diff)
downloadComputeLibrary-ec6997563a7cccf58431267cca39435ecd57cd32.tar.gz
COMPMID-2076: Add StackLayer to the graph API
Change-Id: Ifae23659c2471d9c052bc8adf066c5228d6e8b23 Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-on: https://review.mlplatform.org/c/893 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
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 e23b2b9897..4eb6a0f886 100644
--- a/arm_compute/graph/nodes/Nodes.h
+++ b/arm_compute/graph/nodes/Nodes.h
@@ -55,6 +55,7 @@
#include "arm_compute/graph/nodes/SliceLayerNode.h"
#include "arm_compute/graph/nodes/SoftmaxLayerNode.h"
#include "arm_compute/graph/nodes/SplitLayerNode.h"
+#include "arm_compute/graph/nodes/StackLayerNode.h"
#include "arm_compute/graph/nodes/UpsampleLayerNode.h"
#include "arm_compute/graph/nodes/YOLOLayerNode.h"