22 lines
606 B
C++
22 lines
606 B
C++
//
|
|
// Copyright © 2020 Arm Ltd. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <armnn/TensorFwd.hpp>
|
|
#include <armnn/Types.hpp>
|
|
|
|
namespace armnnUtils
|
|
{
|
|
|
|
armnn::TensorShape TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings);
|
|
|
|
armnn::TensorInfo TransposeTensorShape(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings);
|
|
|
|
void Transpose(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
|
|
const void* src, void* dst, size_t dataTypeSize);
|
|
|
|
} // namespace armnnUtils
|