WorkloadUtils.hpp File Reference

#include "CpuTensorHandle.hpp"
#include <armnn/backends/ITensorHandle.hpp>
#include <armnn/Tensor.hpp>
#include <armnnUtils/Permute.hpp>
#include <Half.hpp>
#include <Profiling.hpp>
#include <boost/cast.hpp>

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2020 ARM Limited.

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 ConstCpuTensorHandle *tensor, const PermutationVector &permutationVector, void *permuteBuffer)
void	ReshapeWeightsForAcl (TensorInfo &weightInfo, DataLayout dataLayout)
TensorInfo	ConvertWeightTensorInfoFromArmnnToAcl (const TensorInfo &weightInfo, DataLayout dataLayout)
armnn::ConstTensor	ConvertWeightTensorFromArmnnToAcl (const ConstCpuTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)