| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.cpp | 19 const PermutationVector& permutationVector, void* permuteBuffer) in PermuteTensor() argument
26 if (permutationVector.GetSize() > 0) in PermuteTensor()
28 tensorInfo = armnnUtils::Permuted(tensorInfo, permutationVector); in PermuteTensor()
29 armnnUtils::Permute(tensorInfo.GetShape(), permutationVector, in PermuteTensor()
127 PermutationVector permutationVector{ 3, 2, 0, 1 }; in ConvertWeightTensorInfoFromArmnnToAcl() local
128 weightPermutedInfo = armnnUtils::Permuted(weightInfo, permutationVector); in ConvertWeightTensorInfoFromArmnnToAcl()
146 PermutationVector permutationVector{}; in Convert1HWOTensorToAcl() local
157 permutationVector = { 0, 2, 3, 1 }; in Convert1HWOTensorToAcl()
165 ConstTensor weightsPermuted = PermuteTensor(weightTensor, permutationVector, permuteBuffer); in Convert1HWOTensorToAcl()
188 PermutationVector permutationVector{ 0, 2, 3, 1 }; in Convert1HWOTensorInfoToAcl() local
[all …]
|
| D | WorkloadUtils.hpp | 209 const PermutationVector& permutationVector,
|
| /external/armnn/src/armnn/optimizations/ |
| D | PermuteDepthwiseConv2dWeights.hpp | 56 PermutationVector permutationVector = { 0, 2, 3, 1 }; in AddPermuteLayer() local
57 TensorInfo weightsPermuted = armnnUtils::Permuted(weightInfo, permutationVector); in AddPermuteLayer()
62 PermuteDescriptor(permutationVector), in AddPermuteLayer()
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.hpp | 165 … armnn::Optional<armnn::PermutationVector&> permutationVector = armnn::EmptyOptional());
169 … armnn::Optional<armnn::PermutationVector&> permutationVector = armnn::EmptyOptional());
|
| D | OnnxParser.cpp | 580 const armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateConstTensorImpl() argument
586 if (permutationVector.has_value() && permutationVector.value().GetSize() > 0) in CreateConstTensorImpl()
588 tensorInfo = armnnUtils::Permuted(tensorInfo, permutationVector.value()); in CreateConstTensorImpl()
589 armnnUtils::Permute(tensorInfo.GetShape(), permutationVector.value(), in CreateConstTensorImpl()
602 armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateConstTensor() argument
636 return CreateConstTensorImpl<float>(srcData, tensorInfo, permutationVector); in CreateConstTensor()
642 permutationVector); in CreateConstTensor()
648 armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateInt64ConstTensor() argument
690 return CreateConstTensorImpl<int32_t>(int32Data.data(), tensorInfo, permutationVector); in CreateInt64ConstTensor()
701 return CreateConstTensorImpl<int32_t>(int32Data.data(), tensorInfo, permutationVector); in CreateInt64ConstTensor()
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.hpp | 270 armnn::Optional<armnn::PermutationVector&> permutationVector);
282 armnn::Optional<armnn::PermutationVector&> permutationVector);
|
| D | TfLiteParser.cpp | 594 armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateConstTensorImpl() argument
603 if (permutationVector.has_value() && permutationVector.value().GetSize() > 0) in CreateConstTensorImpl()
605 tensorInfo = armnnUtils::Permuted(tensorInfo, permutationVector.value()); in CreateConstTensorImpl()
606 armnnUtils::Permute(tensorInfo.GetShape(), permutationVector.value(), in CreateConstTensorImpl()
1565 … PermutationVector permutationVector(permuteShape.data(), permuteTensorInfo.GetNumElements()); in ParseTranspose() local
1567 desc = TransposeDescriptor(permutationVector); in ParseTranspose()
5098 … armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateConstTensorAndStoreData() argument
5106 permutationVector); in CreateConstTensorAndStoreData()
5135 … armnn::Optional<armnn::PermutationVector&> permutationVector) in CreateConstTensorPermuted() argument
5150 permutationVector); in CreateConstTensorPermuted()
[all …]
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.cpp | 1446 PermutationVector permutationVector = { 3, 2, 0, 1 }; in ParseConstant() local
1449 weightsInfo = armnnUtils::Permuted(weightsInfo, permutationVector); in ParseConstant()
1450 armnnUtils::Permute(weightsInfo.GetShape(), permutationVector, in ParseConstant()
1679 PermutationVector permutationVector = { 3, 2, 0, 1 }; in ParseDepthwiseConvolution2d() local
1682 weightsInfo = armnnUtils::Permuted(weightsInfo, permutationVector); in ParseDepthwiseConvolution2d()
1683 armnnUtils::Permute(weightsInfo.GetShape(), permutationVector, in ParseDepthwiseConvolution2d()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | Conv2dTestImpl.cpp | 3734 armnn::PermutationVector permutationVector {3,0,1,2}; in DepthwiseConvolution2dDepthMul64Test() local
3735 kernelTensorInfo = armnnUtils::Permuted(kernelTensorInfo, permutationVector); in DepthwiseConvolution2dDepthMul64Test()
3737 armnnUtils::Permute(kernelTensorInfo.GetShape(), permutationVector, in DepthwiseConvolution2dDepthMul64Test()
|