aboutsummaryrefslogtreecommitdiff
path: root/src/graph/backends/NEON/NENodeValidator.cpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-08-02 13:41:49 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:54:54 +0000
commit57c48243af8fd4b503fe629166531299c1b083d3 (patch)
tree6d48c2681663e3a4a34cf13a725124bfede6fe53 /src/graph/backends/NEON/NENodeValidator.cpp
parentebbb7f2ee00bdebad5da3629bbc78dc3a65fe0c5 (diff)
downloadComputeLibrary-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 'src/graph/backends/NEON/NENodeValidator.cpp')
-rw-r--r--src/graph/backends/NEON/NENodeValidator.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/graph/backends/NEON/NENodeValidator.cpp b/src/graph/backends/NEON/NENodeValidator.cpp
index 58589d5449..58ffaf024b 100644
--- a/src/graph/backends/NEON/NENodeValidator.cpp
+++ b/src/graph/backends/NEON/NENodeValidator.cpp
@@ -57,6 +57,8 @@ Status NENodeValidator::validate(INode *node)
case NodeType::DepthwiseConvolutionLayer:
return detail::validate_depthwise_convolution_layer<NEDepthwiseConvolutionLayer,
NEDepthwiseConvolutionLayer3x3>(*polymorphic_downcast<DepthwiseConvolutionLayerNode *>(node));
+ case NodeType::PermuteLayer:
+ return detail::validate_permute_layer<NEPermute>(*polymorphic_downcast<PermuteLayerNode *>(node));
default:
return Status{};
}