• Home
  • Raw
  • Download

Lines Matching refs:permutationVector

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
189 weightsPermuted = armnnUtils::Permuted(weightInfo, permutationVector); in Convert1HWOTensorInfoToAcl()
224 PermutationVector permutationVector = { 2, 3, 1, 0 }; in Convert1HWOtoMIHW() local
225 ConstTensor weightsPermuted = PermuteTensor(weightTensor, permutationVector, permuteBuffer); in Convert1HWOtoMIHW()
247 PermutationVector permutationVector{}; in ConvertWeightTensorFromArmnnToAcl() local
251 permutationVector = { 3, 2, 0, 1 }; in ConvertWeightTensorFromArmnnToAcl()
253 ConstTensor weightPermuted = PermuteTensor(weightTensor, permutationVector, permuteBuffer); in ConvertWeightTensorFromArmnnToAcl()
346 armnn::PermutationVector permutationVector{}; in GeneratePermutationVectorOnLastTwoDimensions() local
350 permutationVector = {1U, 0U}; in GeneratePermutationVectorOnLastTwoDimensions()
353 permutationVector = {0U, 2U, 1U}; in GeneratePermutationVectorOnLastTwoDimensions()
356 permutationVector = {0U, 1U, 3U, 2U}; in GeneratePermutationVectorOnLastTwoDimensions()
361 return permutationVector; in GeneratePermutationVectorOnLastTwoDimensions()