ArmNN
NotReleased
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,
const
armnn::PermutationVector
& mappings);
15
16
armnn::TensorInfo
Permuted
(
const
armnn::TensorInfo
& info,
const
armnn::PermutationVector
& mappings);
17
18
void
Permute
(
const
armnn::TensorShape
& dstShape,
const
armnn::PermutationVector
& mappings,
19
const
void
* src,
void
* dst,
size_t
dataTypeSize);
20
21
}
// namespace armnnUtils
armnnUtils::Permute
void Permute(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
Definition:
Permute.cpp:121
armnn::PermutationVector
Definition:
Types.hpp:170
armnn::TensorInfo
Definition:
Tensor.hpp:53
armnnUtils::Permuted
armnn::TensorShape Permuted(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)
Definition:
Permute.cpp:98
TensorFwd.hpp
armnnUtils
Definition:
DataLayoutIndexed.hpp:13
Types.hpp
armnn::TensorShape
Definition:
Tensor.hpp:20
include
armnnUtils
Permute.hpp
Generated on Fri Mar 13 2020 16:06:53 for ArmNN by
1.8.13