| /external/armnn/delegate/classic/src/ |
| D | FullyConnected.hpp | 58 const armnn::TensorInfo& weightsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteWeightsTensor); in VisitFullyConnectedOperator() local
76 int32_t weightsDimension = static_cast<int32_t>(weightsTensorInfo.GetNumDimensions()); in VisitFullyConnectedOperator()
106 reshapedDimensions[1] = weightsTensorInfo.GetShape()[1]; in VisitFullyConnectedOperator()
126 reshapedDimensions[1] = weightsTensorInfo.GetShape()[0]; in VisitFullyConnectedOperator()
143 descriptor.m_ConstantWeights = weightsTensorInfo.IsConstant(); in VisitFullyConnectedOperator()
158 weightsTensorInfo, in VisitFullyConnectedOperator()
174 if (weightsTensorInfo.IsConstant()) in VisitFullyConnectedOperator()
177 weightsTensorInfo); in VisitFullyConnectedOperator()
182 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsTensorInfo); in VisitFullyConnectedOperator()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | PermuteDepthwiseConv2dWeightsTests.cpp | 20 const armnn::TensorInfo& weightsTensorInfo, in CreateSimpleDepthwiseConv2dNetwork() argument
31 weightsInputLayer->GetOutputSlot(0).SetTensorInfo(weightsTensorInfo); in CreateSimpleDepthwiseConv2dNetwork()
81 armnn::TensorInfo weightsTensorInfo({ 2, 6 }, armnn::DataType::Float32); in PermuteDepthwiseConv2dWeightsTest() local
88 weightsTensorInfo, in PermuteDepthwiseConv2dWeightsTest()
92 weightsTensorInfo.GetShape(), in PermuteDepthwiseConv2dWeightsTest()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | FullyConnectedEndToEndTestImpl.hpp | 24 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNonConstWeights() argument
35 Connect(weightsInputLayer, fullyConnectedLayer, weightsTensorInfo, 0, 1); in CreateFullyConnectedNetworkNonConstWeights()
43 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNonConstWeightsConstBias() argument
57 Connect(weightsInputLayer, fullyConnectedLayer, weightsTensorInfo, 0, 1); in CreateFullyConnectedNetworkNonConstWeightsConstBias()
66 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkConstWeightsNonConstBias() argument
80 Connect(weightsLayer, fullyConnectedLayer, weightsTensorInfo, 0, 1); in CreateFullyConnectedNetworkConstWeightsNonConstBias()
146 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNoConnectedBiasExplicit() argument
158 Connect(weightsLayer, fullyConnectedLayer, weightsTensorInfo, 0, 1); in CreateFullyConnectedNetworkNoConnectedBiasExplicit()
178 armnn::TensorInfo weightsTensorInfo({ 2, 6 }, ArmnnType); in FullyConnectedWithDynamicWeightsEndToEnd() local
179 weightsTensorInfo.SetQuantizationScale(0.2f); in FullyConnectedWithDynamicWeightsEndToEnd()
[all …]
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 533 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local
534 weightsTensorInfo.SetConstant();
541 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
546 Connect(weights, layer, weightsTensorInfo, 0, 1);
597 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local
598 weightsTensorInfo.SetConstant();
624 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
631 Connect(weights, layer, weightsTensorInfo, 0, 1);
691 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local
692 weightsTensorInfo.SetConstant();
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | FullyConnectedTestImpl.cpp | 30 armnn::TensorInfo weightsTensorInfo, in SimpleFullyConnectedTestImpl() argument
40 …ptr<armnn::ITensorHandle> input1Handle = tensorHandleFactory.CreateTensorHandle(weightsTensorInfo); in SimpleFullyConnectedTestImpl()
49 AddInputToWorkload(data, info, weightsTensorInfo, input1Handle.get()); in SimpleFullyConnectedTestImpl()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 183 const TensorInfo& weightsTensorInfo, in ValidateBiasTensorQuantization() argument
205 …if (biasTensor.HasMultipleQuantizationScales() || weightsTensorInfo.HasMultipleQuantizationScales(… in ValidateBiasTensorQuantization()
208 const std::vector<float>& weightScales = weightsTensorInfo.GetQuantizationScales(); in ValidateBiasTensorQuantization()
229 …const float expectedScale = inputTensorInfo.GetQuantizationScale() * weightsTensorInfo.GetQuantiza… in ValidateBiasTensorQuantization()
|