Home
last modified time | relevance | path

Searched refs:weightsInfo (Results 1 – 25 of 31) sorted by relevance

12

/external/armnn/src/backends/backendsCommon/test/
DLayerReleaseConstantDataTest.cpp89 TensorInfo weightsInfo = TensorInfo({ 2, 3, 5, 3 }, armnn::DataType::Float32, 1.0, 0.0, true); variable
92 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
95 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
138 TensorInfo weightsInfo = TensorInfo({ 3, 3, 5, 3 }, armnn::DataType::Float32, 1.0, 0.0, true); variable
141 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
144 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
185 TensorInfo weightsInfo = TensorInfo({ 7, 20 }, DataType::QAsymmU8, inputsQScale, 0.0, true); variable
188 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
191 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
DConvolution2dEndToEndTestImpl.hpp23 const armnn::TensorInfo& weightsInfo, in CreateConstConvolution2dNetwork() argument
39 Connect(weightsLayer, convolution2d, weightsInfo, 0, 1); in CreateConstConvolution2dNetwork()
64 TensorInfo weightsInfo({ 1, 3, 3, 1 }, ArmnnType, qScale, qOffset, true); in Convolution2dEndToEnd() local
106 PermuteTensorNhwcToNchw(weightsInfo, weightsData); in Convolution2dEndToEnd()
116 ConstTensor weights(weightsInfo, qWeightsData); in Convolution2dEndToEnd()
121 weightsInfo, in Convolution2dEndToEnd()
DConvolution3dEndToEndTestImpl.hpp23 const armnn::TensorInfo& weightsInfo, in CreateConvolution3dNetwork() argument
39 Connect(weightsLayer, convolution3d, weightsInfo, 0, 1); in CreateConvolution3dNetwork()
61 TensorInfo weightsInfo({ 3, 3, 3, 1, 1 }, ArmnnType, qScale, qOffset, true); in Convolution3dEndToEnd() local
152 ConstTensor weights(weightsInfo, qWeightsData); in Convolution3dEndToEnd()
157 weightsInfo, in Convolution3dEndToEnd()
DDepthwiseConvolution2dEndToEndTests.hpp23 const armnn::TensorInfo& weightsInfo, in CreateDepthwiseConvolution2dNetwork() argument
39 Connect(weightsLayer, convolution2d, weightsInfo, 0, 1); in CreateDepthwiseConvolution2dNetwork()
76 …TensorInfo weightsInfo({1, kernelHeight, kernelWidth, outputChannels}, ArmnnType, qScale, qOffset,… in DepthwiseConvolution2dEndToEnd() local
168 ConstTensor weights(weightsInfo, qWeightsData); in DepthwiseConvolution2dEndToEnd()
173 weightsInfo, in DepthwiseConvolution2dEndToEnd()
DTransposeConvolution2dEndToEndTestImpl.hpp70 TensorInfo weightsInfo(weightsShape, ArmnnType, qScale, qOffset, true); in TransposeConvolution2dEndToEnd() local
119 …armnnUtils::Permute(weightsInfo.GetShape(), nchwToNhwc, weightsData.data(), tmp.data(), dataTypeSi… in TransposeConvolution2dEndToEnd()
135 ConstTensor weights(weightsInfo, qWeightsData); in TransposeConvolution2dEndToEnd()
163 TensorInfo weightsInfo({1, 2, 2, 1}, ArmnnType, qScale, qOffset, true); in SimpleTransposeConvolution2dEndToEnd() local
204 ConstTensor weights(weightsInfo, qWeightsData); in SimpleTransposeConvolution2dEndToEnd()
DOptimizedNetworkTests.cpp382 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
385 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements());
386 armnn::ConstTensor weights(weightsInfo, weightsData);
411 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
/external/armnn/src/backends/backendsCommon/test/layerTests/
DTransposeConvolution2dTestImpl.cpp123 armnn::TensorInfo& weightsInfo, in TransposeConvolution2dTest() argument
142 weightsInfo.SetQuantizationScale(qScale); in TransposeConvolution2dTest()
143 weightsInfo.SetQuantizationOffset(qOffset); in TransposeConvolution2dTest()
159 weightsInfo, in TransposeConvolution2dTest()
161 weightsInfo.GetQuantizationScale(), in TransposeConvolution2dTest()
162 weightsInfo.GetQuantizationOffset()) in TransposeConvolution2dTest()
209 armnn::TensorInfo& weightsInfo, in SwizzleData() argument
214 PermuteTensorNchwToNhwc<T>(weightsInfo, weightsData); in SwizzleData()
247 TensorInfo weightsInfo(weightsShape, ArmnnType); in SimpleTransposeConvolution2dTest() local
291 SwizzleData(inputInfo, inputData, outputInfo, expectedOutputData, weightsInfo, weightsData); in SimpleTransposeConvolution2dTest()
[all …]
/external/armnn/src/armnn/test/optimizations/
DConvertConstantsFloatToHalfTests.cpp28 …armnn::TensorInfo weightsInfo = armnn::TensorInfo(4, dims, armnn::DataType::Float32, 0.0f, 0, true… variable
29 armnn::ConstTensor weights(weightsInfo, floatWeights);
40 weightsLayer->GetOutputSlot().SetTensorInfo(weightsInfo);
87 const armnn::TensorInfo weightsInfo({ 4 }, armnn::DataType::Float32, 0.0f, 0, true); variable
91 armnn::ConstTensor weightsTensor(weightsInfo, &expectedWeightsData);
97 weights->GetOutputSlot().SetTensorInfo(weightsInfo);
DFoldPadTests.cpp28 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
49 ConstTensor weights(weightsInfo, weightsVector);
53 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
110 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
131 ConstTensor weights(weightsInfo, weightsVector);
134 weightsLayer->GetOutputSlot().SetTensorInfo(weightsInfo);
691 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
692 ConstTensor weights(weightsInfo, weightsData);
815 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 0.0f, 0, true); variable
816 ConstTensor weights(weightsInfo, weightsData);
DFuseActivationTests.cpp79 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
80 ConstTensor weights(weightsInfo, weightsVector); in AddConstantLayers()
83 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in AddConstantLayers()
131 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
132 ConstTensor weights(weightsInfo, weightsVector); in AddConstantLayers()
135 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in AddConstantLayers()
179 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
180 ConstTensor weights(weightsInfo, weightsVector); in AddConstantLayers()
183 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in AddConstantLayers()
DFuseBatchNormTests.cpp145 TensorInfo weightsInfo(4, weightsDimensionSizes, ArmnnType, 0.0f, 0, true); in CreateNetwork() local
146 ConstTensor weights(weightsInfo, weightsVector); in CreateNetwork()
/external/armnn/samples/
DSimpleSample.cpp31 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
32 weightsInfo.SetConstant(); in main()
33 ConstTensor weights(weightsInfo, weightsData); in main()
58 constantWeightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in main()
DAsyncExecutionSample.cpp52 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
53 weightsInfo.SetConstant(); in main()
54 ConstTensor weights(weightsInfo, weightsData); in main()
79 constantWeightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in main()
DCustomMemoryAllocatorSample.cpp81 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
82 weightsInfo.SetConstant(true); in main()
83 ConstTensor weights(weightsInfo, weightsData); in main()
94 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in main()
/external/armnn/src/backends/backendsCommon/
DWorkloadUtils.cpp144 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOTensorToAcl() local
151 depthMultiplier = weightsInfo.GetShape()[3] / inputInfo.GetShape()[3]; in Convert1HWOTensorToAcl()
156 depthMultiplier = weightsInfo.GetShape()[3] / inputInfo.GetShape()[1]; in Convert1HWOTensorToAcl()
206 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOtoMIHW() local
208 if (weightsInfo.HasPerAxisQuantization()) in Convert1HWOtoMIHW()
215 auto weightsShape = weightsInfo.GetShape(); in Convert1HWOtoMIHW()
218 weightsInfo.SetShape({ weightsShape[1], in Convert1HWOtoMIHW()
/external/armnn/src/backends/reference/workloads/
DRefTransposeConvolution2dWorkload.cpp22 const TensorInfo& weightsInfo = m_Weights->GetTensorInfo(); in RefTransposeConvolution2dWorkload() local
24 m_WeightsDecoder = MakeDecoder<float>(weightsInfo, m_Weights->Map(true)); in RefTransposeConvolution2dWorkload()
25 m_WeightsShape = weightsInfo.GetShape(); in RefTransposeConvolution2dWorkload()
/external/armnn/src/backends/cl/test/
DClCustomAllocatorTests.cpp68 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in CreateTestNetwork() local
69 weightsInfo.SetConstant(true); in CreateTestNetwork()
70 armnn::ConstTensor weights(weightsInfo, weightsData); in CreateTestNetwork()
83 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo); in CreateTestNetwork()
/external/armnn/src/backends/tosaReference/test/
DTosaRefLayerSupportTests.cpp152 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32); variable
161 … {inputInfo, outputInfo, weightsInfo, biasesInfo},
175 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
184 … {inputInfo, outputInfo, weightsInfo, biasesInfo},
511 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32); variable
522 … {inputInfo, outputInfo, weightsInfo, biasesInfo},
535 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
544 … {inputInfo, outputInfo, weightsInfo, biasesInfo},
/external/armnn/src/armnnSerializer/test/
DSerializerTests.cpp421 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
424 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements());
425 armnn::ConstTensor weights(weightsInfo, weightsData);
455 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
462 …Convolution2dLayerVerifier verifier(layerName, {inputInfo, weightsInfo, biasesInfo}, {outputInfo},…
528 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
531 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements());
532 armnn::ConstTensor weights(weightsInfo, weightsData);
563 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
572 layerName, {inputInfo, weightsInfo, biasesInfo}, {outputInfo}, descriptor, constants);
[all …]
/external/armnn/src/backends/tosaCommon/test/
DOneToOneMappingTests.cpp189 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
197 … {&inputInfo, &weightsInfo, &biasesInfo},
225 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
231 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements());
232 ConstTensor weights(weightsInfo, weightsData);
249 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
623 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
638 … {&inputInfo, &weightsInfo, &biasesInfo},
669 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
672 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements());
[all …]
/external/armnn/src/backends/neon/workloads/
DNeonDepthwiseConvolutionWorkload.cpp103 arm_compute::ITensorInfo* weightsInfo = weights.info(); in NeonDepthwiseConvolutionWorkload() local
105 auto weightsShape = weightsInfo->tensor_shape(); in NeonDepthwiseConvolutionWorkload()
/external/armnn/src/backends/cl/workloads/
DClDepthwiseConvolutionWorkload.cpp116 arm_compute::ITensorInfo* weightsInfo = weights.info(); in ClDepthwiseConvolutionWorkload() local
118 auto weightsShape = weightsInfo->tensor_shape(); in ClDepthwiseConvolutionWorkload()
/external/armnn/src/armnnDeserializer/
DDeserializer.cpp1447 armnn::TensorInfo weightsInfo = input.GetInfo(); in ParseConstant() local
1448 … std::unique_ptr<unsigned char[]> permuteBuffer(new unsigned char[weightsInfo.GetNumBytes()]); in ParseConstant()
1449 weightsInfo = armnnUtils::Permuted(weightsInfo, permutationVector); in ParseConstant()
1450 armnnUtils::Permute(weightsInfo.GetShape(), permutationVector, in ParseConstant()
1452 GetDataTypeSize(weightsInfo.GetDataType())); in ParseConstant()
1455 auto weightsShape = weightsInfo.GetShape(); in ParseConstant()
1456 weightsInfo.SetShape({1, in ParseConstant()
1460 weightsInfo.SetConstant(true); in ParseConstant()
1462 armnn::ConstTensor weightsPermuted(weightsInfo, permuteBuffer.get()); in ParseConstant()
1680 armnn::TensorInfo weightsInfo = weights.GetInfo(); in ParseDepthwiseConvolution2d() local
[all …]
/external/android-nn-driver/
DConversionUtils.hpp2391 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertConv2d() local
2422 const uint32_t kernelX = weightsInfo.GetShape()[2]; in ConvertConv2d()
2423 const uint32_t kernelY = weightsInfo.GetShape()[1]; in ConvertConv2d()
2450 weightsInfo, in ConvertConv2d()
2630 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertDepthwiseConv2d() local
2659 const uint32_t kernelX = weightsInfo.GetShape()[2]; in ConvertDepthwiseConv2d()
2660 const uint32_t kernelY = weightsInfo.GetShape()[1]; in ConvertDepthwiseConv2d()
2687 weightsInfo, in ConvertDepthwiseConv2d()
3144 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertFullyConnected() local
3148 … reshapedInfo.SetShape(FlattenFullyConnectedInput(inputInfo.GetShape(), weightsInfo.GetShape())); in ConvertFullyConnected()
[all …]
DConversionUtils_1_2.hpp439 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertConv2d_1_2() local
459 const uint32_t kernelX = weightsInfo.GetShape()[widthIndex]; in ConvertConv2d_1_2()
460 const uint32_t kernelY = weightsInfo.GetShape()[heightIndex]; in ConvertConv2d_1_2()
503 weightsInfo, in ConvertConv2d_1_2()
612 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertDepthwiseConv2d_1_2() local
629 const uint32_t kernelX = weightsInfo.GetShape()[2]; in ConvertDepthwiseConv2d_1_2()
630 const uint32_t kernelY = weightsInfo.GetShape()[1]; in ConvertDepthwiseConv2d_1_2()
672 weightsInfo, in ConvertDepthwiseConv2d_1_2()
1179 const TensorInfo& weightsInfo = weights.GetInfo(); in ConvertGroupedConv2d() local
1180 TensorInfo groupWeightsInfo(weightsInfo); in ConvertGroupedConv2d()
[all …]

12