Home
last modified time | relevance | path

Searched refs:weightTensor (Results 1 – 4 of 4) sorted by relevance

/external/armnn/src/backends/backendsCommon/
DWorkloadUtils.cpp139 std::tuple<ConstTensor, unsigned int> Convert1HWOTensorToAcl(const ConstTensorHandle* weightTensor, in Convert1HWOTensorToAcl() argument
144 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOTensorToAcl()
165 ConstTensor weightsPermuted = PermuteTensor(weightTensor, permutationVector, permuteBuffer); in Convert1HWOTensorToAcl()
201 std::tuple<ConstTensor, unsigned int> Convert1HWOtoMIHW(const ConstTensorHandle* weightTensor, in Convert1HWOtoMIHW() argument
206 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOtoMIHW()
225 ConstTensor weightsPermuted = PermuteTensor(weightTensor, permutationVector, permuteBuffer); in Convert1HWOtoMIHW()
230 armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstTensorHandle* weightTensor, in ConvertWeightTensorFromArmnnToAcl() argument
234 ARMNN_ASSERT_MSG(weightTensor, "Invalid input tensor"); in ConvertWeightTensorFromArmnnToAcl()
237 auto multiplier = weightTensor->GetTensorInfo().GetShape()[0]; in ConvertWeightTensorFromArmnnToAcl()
238 auto inputChannels = weightTensor->GetTensorInfo().GetShape()[1]; in ConvertWeightTensorFromArmnnToAcl()
[all …]
DWorkloadUtils.hpp224 armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstTensorHandle* weightTensor,
237 std::tuple<ConstTensor, unsigned int> Convert1HWOTensorToAcl(const ConstTensorHandle* weightTensor,
249 std::tuple<ConstTensor, unsigned int> Convert1HWOtoMIHW(const ConstTensorHandle* weightTensor,
/external/armnn/src/backends/backendsCommon/test/layerTests/
DTransposeConvolution2dTestImpl.cpp647 ScopedTensorHandle weightTensor(kernelInfo); in TransposeConvolution2dPerAxisQuantTest() local
650 AllocateAndCopyDataToITensorHandle(&weightTensor, kernelData.data()); in TransposeConvolution2dPerAxisQuantTest()
655 queueDescriptor.m_Weight = &weightTensor; in TransposeConvolution2dPerAxisQuantTest()
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1099 auto weightTensor = CreateConstTensor(node.input(1)); in AddConvLayerWithDepthwiseConv() local
1100 IConnectableLayer* weightsLayer = m_Network->AddConstantLayer(weightTensor.first); in AddConvLayerWithDepthwiseConv()
1104 TensorInfo weightsPermuted = armnnUtils::Permuted(weightTensor.first.GetInfo(), perVec); in AddConvLayerWithDepthwiseConv()
1112 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightTensor.first.GetInfo()); in AddConvLayerWithDepthwiseConv()
1877 auto weightTensor = CreateConstTensor(node.input(1)); in ParseConv() local
1879 IConnectableLayer* weightsLayer = m_Network->AddConstantLayer(weightTensor.first); in ParseConv()
1880 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightTensor.first.GetInfo()); in ParseConv()