21.05
|
#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) |