mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-20 22:25:54 +00:00
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
|