WorkloadUtils.hpp File Reference

#include <armnn/backends/ITensorHandle.hpp>
#include <armnn/backends/TensorHandle.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/utility/PolymorphicDowncast.hpp>
#include <armnnUtils/Permute.hpp>
#include <Half.hpp>
#include <Profiling.hpp>

Include dependency graph for WorkloadUtils.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Functions
template<typename CopyFunc>
void	CopyTensorContentsGeneric (const ITensorHandle *srcTensor, ITensorHandle *dstTensor, CopyFunc copy)
template<typename SrcTensorHandleType, typename DstTensorHandleType, typename DescriptorType>
void	GatherTensorHandlePairs (const DescriptorType &descriptor, std::vector< std::pair< SrcTensorHandleType *, DstTensorHandleType * > > &tensorHandlePairs)
int32_t	ConvertMaskToACLFormat (int32_t mask, int32_t numDim)
armnn::ConstTensor	PermuteTensor (const ConstTensorHandle *tensor, const PermutationVector &permutationVector, void *permuteBuffer)
void	ReshapeWeightsForAcl (TensorInfo &weightInfo, DataLayout dataLayout)
TensorInfo	ConvertWeightTensorInfoFromArmnnToAcl (const TensorInfo &weightInfo, DataLayout dataLayout)
std::tuple< TensorInfo, unsigned int >	Convert1HWOTensorInfoToAcl (const TensorInfo &weightInfo, const TensorInfo &inputInfo, const DataLayout dataLayout)
	Weights for depthwise have a datalayout of [1,H,W,O] = [1,H,W,I*M] This function coverts a TensorInfo from [1,H,W,I*M] to [1,I*M,H,W] (if NCHW) or keeps it at [1,H,W,I*M] (if NHWC) as required by the compute library Returns a tuple of converted weights tensor info and depth multiplier.
armnn::ConstTensor	ConvertWeightTensorFromArmnnToAcl (const ConstTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)
std::tuple< ConstTensor, unsigned int >	Convert1HWOTensorToAcl (const ConstTensorHandle *weightTensor, const TensorInfo &inputInfo, const DataLayout dataLayout, void *permuteBuffer)
	Weights for depthwise have a datalayout of [1,H,W,O] = [1,H,W,I*M] This function coverts a ConstCpuTensorHandle from [1,H,W,I*M] to [1,I*M,H,W] (if NCHW) or keeps it at [1,H,W,I*M] (if NHWC) as required by the compute library.
std::tuple< ConstTensor, unsigned int >	Convert1HWOtoMIHW (const ConstTensorHandle *weightTensor, const TensorInfo &inputInfo, const DataLayout &dataLayout, void *permuteBuffer)
	Converts a (weights) tensor from [1, H, W, I*M] = [1, H, W, O] to [M, I, H, W].
std::map< std::string, unsigned int >	CalculateGatherNdKeyIndices (TensorInfo inputInfo0, TensorInfo inputInfo1)
	Calculates the key index values needed for GatherNd: N, ND, K, W, C (N is always 1)
armnn::PermutationVector	GeneratePermutationVectorOnLastTwoDimensions (unsigned int rank)
	Generates a permutation vector of size rank that permutes the 2 most right dimensions.
std::set< unsigned int >	ComputeSplitAxis (const armnn::SplitterDescriptor &desc, const TensorShape &input)
	Calculates the axis values for split operation.