25.11
Loading...
Searching...
No Matches
Transpose.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020,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
#include <stddef.h>
11
12
namespace
armnnUtils
13
{
14
15
armnn::TensorShape
TransposeTensorShape
(
const
armnn::TensorShape
& srcShape,
const
armnn::PermutationVector
& mappings);
16
17
armnn::TensorInfo
TransposeTensorShape
(
const
armnn::TensorInfo
&
info
,
const
armnn::PermutationVector
& mappings);
18
19
void
Transpose
(
const
armnn::TensorShape
& dstShape,
const
armnn::PermutationVector
& mappings,
20
const
void
* src,
void
* dst,
size_t
dataTypeSize);
21
22
}
// namespace armnnUtils
TensorFwd.hpp
Types.hpp
armnn::PermutationVector
Definition
Types.hpp:317
armnn::TensorInfo
Definition
Tensor.hpp:153
armnn::TensorShape
Definition
Tensor.hpp:21
armnn::BoostLogSeverityMapping::info
@ info
Definition
Logging.hpp:200
armnn::LayerType::Transpose
@ Transpose
Definition
Types.hpp:496
armnnUtils
Definition
CompatibleTypes.hpp:11
armnnUtils::TransposeTensorShape
armnn::TensorShape TransposeTensorShape(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)
Definition
Transpose.cpp:125
include
armnnUtils
Transpose.hpp
Generated on
for Arm NN by
1.14.0