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 /src/graph/GraphBuilder.cpp | |
| 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 'src/graph/GraphBuilder.cpp')
| -rw-r--r-- | src/graph/GraphBuilder.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp index 007f69177a..996af1dc90 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -471,6 +471,11 @@ NodeID GraphBuilder::add_normalization_node(Graph &g, NodeParams params, NodeIdx return create_simple_single_input_output_node<NormalizationLayerNode>(g, params, input, norm_info); } +NodeID GraphBuilder::add_permute_node(Graph &g, NodeParams params, NodeIdxPair input, PermutationVector perm, DataLayout layout) +{ + return create_simple_single_input_output_node<PermuteLayerNode>(g, params, input, perm, layout); +} + NodeID GraphBuilder::add_pooling_node(Graph &g, NodeParams params, NodeIdxPair input, PoolingLayerInfo pool_info) { return create_simple_single_input_output_node<PoolingLayerNode>(g, params, input, pool_info); |