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