hmz007
36ed224bac
Signed-off-by: hmz007 <hmz007@gmail.com> Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
61 lines
1.8 KiB
C++
61 lines
1.8 KiB
C++
//
|
|
// Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <armnn/Tensor.hpp>
|
|
#include <armnn/Types.hpp>
|
|
|
|
#include <armnnUtils/Permute.hpp>
|
|
|
|
template<typename T>
|
|
void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData)
|
|
{
|
|
const armnn::PermutationVector nchwToNhwc = { 0, 3, 1, 2 };
|
|
|
|
tensorInfo = armnnUtils::Permuted(tensorInfo, nchwToNhwc);
|
|
|
|
std::vector<T> tmp(tensorData.size());
|
|
armnnUtils::Permute(tensorInfo.GetShape(), nchwToNhwc, tensorData.data(), tmp.data(), sizeof(T));
|
|
tensorData = tmp;
|
|
}
|
|
|
|
template<typename T>
|
|
void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData)
|
|
{
|
|
const armnn::PermutationVector nhwcToNchw = { 0, 2, 3, 1 };
|
|
|
|
tensorInfo = armnnUtils::Permuted(tensorInfo, nhwcToNchw);
|
|
|
|
std::vector<T> tmp(tensorData.size());
|
|
armnnUtils::Permute(tensorInfo.GetShape(), nhwcToNchw, tensorData.data(), tmp.data(), sizeof(T));
|
|
|
|
tensorData = tmp;
|
|
}
|
|
|
|
template<typename T>
|
|
void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData)
|
|
{
|
|
const armnn::PermutationVector ndhwcToNcdhw = { 0, 2, 3, 4, 1 };
|
|
|
|
tensorInfo = armnnUtils::Permuted(tensorInfo, ndhwcToNcdhw);
|
|
|
|
std::vector<T> tmp(tensorData.size());
|
|
armnnUtils::Permute(tensorInfo.GetShape(), ndhwcToNcdhw, tensorData.data(), tmp.data(), sizeof(T));
|
|
tensorData = tmp;
|
|
}
|
|
|
|
template<typename T>
|
|
void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData)
|
|
{
|
|
const armnn::PermutationVector ncdhwToNdhwc = { 0, 4, 1, 2, 3 };
|
|
|
|
tensorInfo = armnnUtils::Permuted(tensorInfo, ncdhwToNdhwc);
|
|
|
|
std::vector<T> tmp(tensorData.size());
|
|
armnnUtils::Permute(tensorInfo.GetShape(), ncdhwToNdhwc, tensorData.data(), tmp.data(), sizeof(T));
|
|
tensorData = tmp;
|
|
}
|