diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-02 13:41:49 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 57c48243af8fd4b503fe629166531299c1b083d3 (patch) | |
| tree | 6d48c2681663e3a4a34cf13a725124bfede6fe53 /arm_compute/graph/Types.h | |
| parent | ebbb7f2ee00bdebad5da3629bbc78dc3a65fe0c5 (diff) | |
| download | ComputeLibrary-57c48243af8fd4b503fe629166531299c1b083d3.tar.gz | |
MLCE-37: Adds PermuteNode support in graph
Change-Id: Iaa93a497e7913c27f2fd09e974125cda5f04bc4b
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/142463
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/Types.h')
| -rw-r--r-- | arm_compute/graph/Types.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h index ee0bf429ad..2df9998fcf 100644 --- a/arm_compute/graph/Types.h +++ b/arm_compute/graph/Types.h @@ -44,6 +44,7 @@ using arm_compute::DataLayout; using arm_compute::DataLayoutDimension; using arm_compute::TensorShape; using arm_compute::Size2D; +using arm_compute::PermutationVector; using arm_compute::ActivationLayerInfo; using arm_compute::NormType; @@ -140,6 +141,7 @@ enum class NodeType FlattenLayer, FullyConnectedLayer, NormalizationLayer, + PermuteLayer, PoolingLayer, ReshapeLayer, ResizeLayer, |