WorkloadUtils.cpp File Reference

#include <backendsCommon/WorkloadUtils.hpp>
#include <armnn/Utils.hpp>
#include <armnn/utility/NumericCast.hpp>

Go to the source code of this file.

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

Functions
armnn::ConstTensor	PermuteTensor (const ConstTensorHandle *tensor, const PermutationVector &permutationVector, void *permuteBuffer)
void	ReshapeWeightsForAcl (TensorInfo &weightInfo, DataLayout dataLayout)
template<typename DataType >
ConstTensor	ReorderWeightChannelsForAcl (const ConstTensor &weightHandle, DataLayout dataLayout, void *permuteBuffer)
TensorInfo	ConvertWeightTensorInfoFromArmnnToAcl (const TensorInfo &weightInfo, DataLayout dataLayout)
armnn::ConstTensor	ConvertWeightTensorFromArmnnToAcl (const ConstTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)
int32_t	ConvertMaskToACLFormat (int32_t mask, int32_t numDim)