From e8f05da5fb919aa209e1bf0e5c70dd15fff84b7f Mon Sep 17 00:00:00 2001 From: thecha01 Date: Mon, 24 Aug 2020 17:21:41 +0100 Subject: Add ArgMinMax layer node to Graph API Change-Id: I2ccb2c65edd2932b76e905af3d747324b65c2f7f Signed-off-by: thecha01 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3910 Comments-Addressed: Arm Jenkins Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins --- arm_compute/graph/nodes/Nodes.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/graph/nodes/Nodes.h') diff --git a/arm_compute/graph/nodes/Nodes.h b/arm_compute/graph/nodes/Nodes.h index 0e4ccd067d..2b400a9140 100644 --- a/arm_compute/graph/nodes/Nodes.h +++ b/arm_compute/graph/nodes/Nodes.h @@ -25,6 +25,7 @@ #define ARM_COMPUTE_GRAPH_NODES_H #include "arm_compute/graph/nodes/ActivationLayerNode.h" +#include "arm_compute/graph/nodes/ArgMinMaxLayerNode.h" #include "arm_compute/graph/nodes/BatchNormalizationLayerNode.h" #include "arm_compute/graph/nodes/BoundingBoxTransformLayerNode.h" #include "arm_compute/graph/nodes/ChannelShuffleLayerNode.h" -- cgit v1.2.1