diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-12-06 20:43:06 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-12-07 15:40:28 +0000 |
| commit | 6bc85258b0a87cde2e1e644d930ea546b6c3a1cc (patch) | |
| tree | 463e31a280a8a57b4d43acd9adb1e05de000b4c0 /src/armnn | |
| parent | 4002a609b1e5a75d7527cd8b7cb05ed444d27caa (diff) | |
| download | armnn-6bc85258b0a87cde2e1e644d930ea546b6c3a1cc.tar.gz | |
IVGCVSW-6853 Rewrite BuildArmComputePermutationVector()
* Some pemutation vectors were not converted correctly.
* Add Transpose end to end test.
* Comments added with an example to clarify the differences betweeen
Transpose and Permute
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6c0954ca6ce00ef5f2a6f3625abe6f4fd27b5cdf
Diffstat (limited to 'src/armnn')
| -rw-r--r-- | src/armnn/layers/PermuteLayer.hpp | 4 | ||||
| -rw-r--r-- | src/armnn/layers/TransposeLayer.hpp | 3 |
2 files changed, 2 insertions, 5 deletions
diff --git a/src/armnn/layers/PermuteLayer.hpp b/src/armnn/layers/PermuteLayer.hpp index 37ae444199..530042515a 100644 --- a/src/armnn/layers/PermuteLayer.hpp +++ b/src/armnn/layers/PermuteLayer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -16,7 +16,6 @@ class PermuteLayer : public LayerWithParameters<PermuteDescriptor> { public: /// Makes a workload for the Permute type. - /// @param [in] graph The graph where this layer can be found. /// @param [in] factory The workload factory which will create the workload. /// @return A pointer to the created workload, or nullptr if not created. virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override; @@ -27,7 +26,6 @@ public: /// Check if the input tensor shape(s) /// will lead to a valid configuration of @ref PermuteLayer. - /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated. void ValidateTensorShapesFromInputs() override; /// By default returns inputShapes if the number of inputs are equal to number of outputs, diff --git a/src/armnn/layers/TransposeLayer.hpp b/src/armnn/layers/TransposeLayer.hpp index 08268f2a54..cd54df5fe1 100644 --- a/src/armnn/layers/TransposeLayer.hpp +++ b/src/armnn/layers/TransposeLayer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020,2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -26,7 +26,6 @@ public: /// Check if the input tensor shape(s) /// will lead to a valid configuration of @ref TransposeLayer. - /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated. void ValidateTensorShapesFromInputs() override; /// Infers the output shapes from given input shapes and the permutation vector. |