blob: b0c1f08a98dd43c925360102c283d65379961013 [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +01001//
Colm Donelana98e79a2022-12-06 21:32:29 +00002// Copyright © 2020,2022 Arm Ltd and Contributors. All rights reserved.
Mike Kellyc9ea45a2020-02-28 18:11:58 +00003// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include <armnn/TensorFwd.hpp>
9#include <armnn/Types.hpp>
Colm Donelana98e79a2022-12-06 21:32:29 +000010#include <stddef.h>
Mike Kellyc9ea45a2020-02-28 18:11:58 +000011
12namespace armnnUtils
13{
14
15armnn::TensorShape TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings);
16
17armnn::TensorInfo TransposeTensorShape(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings);
18
19void Transpose(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
20 const void* src, void* dst, size_t dataTypeSize);
21
22} // namespace armnnUtils