22.02
|
Go to the source code of this file.
Functions | |
template<typename T > | |
void | PermuteTensorNchwToNhwc (armnn::TensorInfo &tensorInfo, std::vector< T > &tensorData) |
template<typename T > | |
void | PermuteTensorNhwcToNchw (armnn::TensorInfo &tensorInfo, std::vector< T > &tensorData) |
template<typename T > | |
void | PermuteTensorNdhwcToNcdhw (armnn::TensorInfo &tensorInfo, std::vector< T > &tensorData) |
template<typename T > | |
void | PermuteTensorNcdhwToNdhwc (armnn::TensorInfo &tensorInfo, std::vector< T > &tensorData) |
void PermuteTensorNcdhwToNdhwc | ( | armnn::TensorInfo & | tensorInfo, |
std::vector< T > & | tensorData | ||
) |
Definition at line 51 of file DataLayoutUtils.hpp.
References TensorInfo::GetShape(), armnnUtils::Permute(), and armnnUtils::Permuted().
void PermuteTensorNchwToNhwc | ( | armnn::TensorInfo & | tensorInfo, |
std::vector< T > & | tensorData | ||
) |
Definition at line 14 of file DataLayoutUtils.hpp.
References TensorInfo::GetShape(), armnnUtils::Permute(), and armnnUtils::Permuted().
Referenced by TransposeConvolution2dPerAxisQuantTest().
void PermuteTensorNdhwcToNcdhw | ( | armnn::TensorInfo & | tensorInfo, |
std::vector< T > & | tensorData | ||
) |
Definition at line 39 of file DataLayoutUtils.hpp.
References TensorInfo::GetShape(), armnnUtils::Permute(), and armnnUtils::Permuted().
Referenced by Convolution3dEndToEnd(), and SimpleConvolution3dTestImpl().
void PermuteTensorNhwcToNchw | ( | armnn::TensorInfo & | tensorInfo, |
std::vector< T > & | tensorData | ||
) |
Definition at line 26 of file DataLayoutUtils.hpp.
References TensorInfo::GetShape(), armnnUtils::Permute(), and armnnUtils::Permuted().
Referenced by Convolution2dPerAxisQuantTest(), and DepthwiseConvolution2dPerAxisQuantTest().