aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/nodes/NodesFwd.h
diff options
context:
space:
mode:
authorthecha01 <theo.charalambous@arm.com>2020-08-28 18:40:38 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2020-09-04 15:54:46 +0000
commit0a05e6a051e6d9028131de10f00b80bbedf44097 (patch)
tree4821a3efd2479e0a770442a30aad10e7db6b1fa0 /arm_compute/graph/nodes/NodesFwd.h
parentbe058f4b5fb2383e0fa4666a92c43ca994486c86 (diff)
downloadComputeLibrary-0a05e6a051e6d9028131de10f00b80bbedf44097.tar.gz
Add DepthToSpace layer node to Graph API
Change-Id: I5db780fb9c94160130e9986bbfc739124bfa8041 Signed-off-by: thecha01 <theo.charalambous@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3914 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
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 9541f4babe..63632848af 100644
--- a/arm_compute/graph/nodes/NodesFwd.h
+++ b/arm_compute/graph/nodes/NodesFwd.h
@@ -38,6 +38,7 @@ class ConcatenateLayerNode;
class ConstNode;
class ConvolutionLayerNode;
class DeconvolutionLayerNode;
+class DepthToSpaceLayerNode;
class DepthwiseConvolutionLayerNode;
class DequantizationLayerNode;
class DetectionOutputLayerNode;