ArmNN
 24.02
Permute.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2019,2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <armnn/TensorFwd.hpp>
9 #include <armnn/Types.hpp>
10 
11 #include <stddef.h>
12 
13 namespace armnnUtils
14 {
15 
17  const armnn::PermutationVector& mappings);
18 
20  const armnn::PermutationVector& mappings);
21 
22 void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
23  const void* src, void* dst, size_t dataTypeSize);
24 
25 } // namespace armnnUtils
armnn::TensorInfo
Definition: Tensor.hpp:152
armnnUtils::Permute
void Permute(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
Definition: Permute.cpp:164
armnnUtils::Permuted
armnn::TensorShape Permuted(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)
Definition: Permute.cpp:125
armnn::TensorShape
Definition: Tensor.hpp:20
TensorFwd.hpp
armnnUtils
Definition: CompatibleTypes.hpp:10
armnn::PermutationVector
Definition: Types.hpp:314
Types.hpp