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.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/arm_compute/graph/nodes/NodesFwd.h b/arm_compute/graph/nodes/NodesFwd.h
index b90cb5c308..05979d796c 100644
--- a/arm_compute/graph/nodes/NodesFwd.h
+++ b/arm_compute/graph/nodes/NodesFwd.h
@@ -32,10 +32,13 @@ namespace graph
class INode;
class ActivationLayerNode;
class BatchNormalizationLayerNode;
+class ChannelShuffleLayerNode;
class ConstNode;
class ConvolutionLayerNode;
+class DeconvolutionLayerNode;
class DepthConcatenateLayerNode;
class DepthwiseConvolutionLayerNode;
+class DummyNode;
class EltwiseLayerNode;
class FlattenLayerNode;
class FullyConnectedLayerNode;
@@ -44,6 +47,7 @@ class NormalizationLayerNode;
class OutputNode;
class PoolingLayerNode;
class ReshapeLayerNode;
+class ResizeLayerNode;
class SoftmaxLayerNode;
class SplitLayerNode;
} // namespace graph