|
| 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. More...
|
| |
| 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. More...
|
| |
| 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]. More...
|
| |
| 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) More...
|
| |
| armnn::PermutationVector | GeneratePermutationVectorOnLastTwoDimensions (unsigned int rank) |
| | Generates a permutation vector of size rank that permutes the 2 most right dimensions. More...
|
| |
| std::set< unsigned int > | ComputeSplitAxis (const armnn::SplitterDescriptor &desc, const TensorShape &input) |
| | Calculates the axis values for split operation. More...
|
| |
1.8.17