20.11
Permute.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2019 Arm Ltd. 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
namespace
armnnUtils
12
{
13
14
armnn::TensorShape
Permuted
(
const
armnn::TensorShape
& srcShape,
15
const
armnn::PermutationVector
& mappings);
16
17
armnn::TensorInfo
Permuted
(
const
armnn::TensorInfo
& info,
18
const
armnn::PermutationVector
& mappings,
19
bool
perChannelPermute =
false
);
20
21
void
Permute
(
const
armnn::TensorShape
& dstShape,
const
armnn::PermutationVector
& mappings,
22
const
void
* src,
void
* dst,
size_t
dataTypeSize);
23
24
}
// namespace armnnUtils
armnn::TensorInfo
Definition:
Tensor.hpp:152
armnn::TensorShape
Definition:
Tensor.hpp:20
TensorFwd.hpp
armnnUtils::Permute
void Permute(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
Definition:
Permute.cpp:131
Types.hpp
armnn::PermutationVector
Definition:
Types.hpp:204
armnnUtils
Definition:
DataLayoutIndexed.hpp:13
armnnUtils::Permuted
armnn::TensorShape Permuted(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)
Definition:
Permute.cpp:98
include
armnnUtils
Permute.hpp
Generated on Wed Nov 25 2020 09:59:10 for ArmNN by
1.8.13