diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-16 15:52:35 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:52:54 +0000 |
commit | 087eaf67dc4be4234a7fcfc3b109c1e4f5e7dd5e (patch) | |
tree | 5cb63112eb567bbc683dfb9a0f51ce4960cf800d /arm_compute/graph/Types.h | |
parent | 1b993389a3ac0cd1b0edc0b11e92fbdee127576f (diff) | |
download | ComputeLibrary-087eaf67dc4be4234a7fcfc3b109c1e4f5e7dd5e.tar.gz |
COMPMID-1176: Adds nodes to the graph.
Nodes added:
-ChannelShuffle
-Resize
-Deconvolution
-Dummy (used for performance analysis and debugging)
Change-Id: Iad19960cbbce6e25532f77bfd34b2292c0ca9781
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/131672
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/graph/Types.h')
-rw-r--r-- | arm_compute/graph/Types.h | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h index d4e4f99377..9cfede9cf3 100644 --- a/arm_compute/graph/Types.h +++ b/arm_compute/graph/Types.h @@ -52,6 +52,7 @@ using arm_compute::PadStrideInfo; using arm_compute::PoolingLayerInfo; using arm_compute::PoolingType; using arm_compute::DimensionRoundingType; +using arm_compute::InterpolationPolicy; /** TODO (geopin01): Make ids strongly typed */ using TensorID = unsigned int; @@ -128,7 +129,9 @@ enum class NodeType { ActivationLayer, BatchNormalizationLayer, + ChannelShuffleLayer, ConvolutionLayer, + DeconvolutionLayer, DepthConcatenateLayer, DepthwiseConvolutionLayer, EltwiseLayer, @@ -137,13 +140,15 @@ enum class NodeType NormalizationLayer, PoolingLayer, ReshapeLayer, - ScaleLayer, + ResizeLayer, SoftmaxLayer, SplitLayer, Input, Output, Const, + + Dummy }; /** Backend Memory Manager affinity **/ |