From 6bc85258b0a87cde2e1e644d930ea546b6c3a1cc Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Tue, 6 Dec 2022 20:43:06 +0000 Subject: 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 Change-Id: I6c0954ca6ce00ef5f2a6f3625abe6f4fd27b5cdf --- src/armnn/layers/PermuteLayer.hpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'src/armnn/layers/PermuteLayer.hpp') 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 { 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 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, -- cgit v1.2.1