You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
648 B
23 lines
648 B
//
|
|
// Copyright © 2020,2022 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <armnn/TensorFwd.hpp>
|
|
#include <armnn/Types.hpp>
|
|
#include <stddef.h>
|
|
|
|
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
|