Home
last modified time | relevance | path

Searched refs:biasTensor (Results 1 – 13 of 13) sorted by relevance

/external/armnn/src/armnnTfLiteParser/test/
DFullyConnected.cpp24 std::string biasTensor = ""; in FullyConnectedFixture() local
29 biasTensor = R"( in FullyConnectedFixture()
34 "name": "biasTensor", in FullyConnectedFixture()
86 }, )" + biasTensor + R"( in FullyConnectedFixture()
205 std::string biasTensor = ""; in FullyConnectedNonConstWeightsFixture() local
211 biasTensor = R"( in FullyConnectedNonConstWeightsFixture()
265 )" + biasTensor + R"( in FullyConnectedNonConstWeightsFixture()
DDepthwiseConvolution2D.cpp23 std::string biasTensor = ""; in DepthwiseConvolution2dFixture() local
28 biasTensor = R"( in DepthwiseConvolution2dFixture()
33 "name": "biasTensor", in DepthwiseConvolution2dFixture()
85 }, )" + biasTensor + R"( in DepthwiseConvolution2dFixture()
239 std::string biasTensor = ""; in DepthwiseConvolution2dFixture2() local
244 biasTensor = R"( in DepthwiseConvolution2dFixture2()
249 "name": "biasTensor", in DepthwiseConvolution2dFixture2()
311 }, )" + biasTensor + R"( in DepthwiseConvolution2dFixture2()
DConv3D.cpp177 "name": "biasTensor", in Conv3DWithBiasesFixture()
DTransposeConv.cpp190 "name": "biasTensor", in TransposeConvFixtureWithBias()
DConv2D.cpp169 "name": "biasTensor", in Conv2DWithBiasesFixture()
/external/armnn/src/armnn/optimizations/
DFuseBatchNorm.hpp133 ConstTensor biasTensor; in Run() local
140 biasTensor = ConstTensor(biasLayer->m_LayerOutput->GetTensorInfo(), in Run()
143 const auto* biasBuffer = static_cast<const T*>(biasTensor.GetMemoryArea()); in Run()
144 std::vector<T> biasVector(biasBuffer, biasBuffer + biasTensor.GetNumElements()); in Run()
/external/armnn/delegate/classic/src/
DFullyConnected.hpp190 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, in VisitFullyConnectedOperator() local
193 … armnn::IConnectableLayer* biasLayer = delegateData.m_Network->AddConstantLayer(biasTensor); in VisitFullyConnectedOperator()
DConvolution.hpp200 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, biasTensorInfo); in VisitConv2dOperator() local
201 … armnn::IConnectableLayer* biasLayer = delegateData.m_Network->AddConstantLayer(biasTensor); in VisitConv2dOperator()
608 auto biasTensor = CreateConstTensor(&tfLiteBiasTensor, biasTensorInfo); in VisitDepthwiseConv2dOperator() local
609 … armnn::IConnectableLayer* biasLayer = delegateData.m_Network->AddConstantLayer(biasTensor); in VisitDepthwiseConv2dOperator()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DTransposeConvolution2dTestImpl.cpp648 ScopedTensorHandle biasTensor(biasInfo); in TransposeConvolution2dPerAxisQuantTest() local
651 AllocateAndCopyDataToITensorHandle(&biasTensor, biasData.data()); in TransposeConvolution2dPerAxisQuantTest()
656 queueDescriptor.m_Bias = &biasTensor; in TransposeConvolution2dPerAxisQuantTest()
DConv2dTestImpl.cpp1811 armnn::ScopedTensorHandle biasTensor(biasDesc); in DepthwiseConvolution2dAsymmetricTestImpl() local
1814 AllocateAndCopyDataToITensorHandle(&biasTensor, bias.data()); in DepthwiseConvolution2dAsymmetricTestImpl()
1982 armnn::ScopedTensorHandle biasTensor(biasDesc); in DepthwiseConvolution2dDepthMul1TestImpl() local
1985 AllocateAndCopyDataToITensorHandle(&biasTensor, biasV.data()); in DepthwiseConvolution2dDepthMul1TestImpl()
2199 armnn::ScopedTensorHandle biasTensor(biasDesc); in DepthwiseConvolution2dTestImpl() local
2202 AllocateAndCopyDataToITensorHandle(&biasTensor, biasV.data()); in DepthwiseConvolution2dTestImpl()
2383 armnn::ScopedTensorHandle biasTensor(biasDesc); in DepthwiseConvolution2dTestImpl() local
2386 AllocateAndCopyDataToITensorHandle(&biasTensor, bias.data()); in DepthwiseConvolution2dTestImpl()
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1126 auto biasTensor = CreateConstTensor(node.input(2)); in AddConvLayerWithDepthwiseConv() local
1129 IConnectableLayer* biasLayer = m_Network->AddConstantLayer(biasTensor.first); in AddConvLayerWithDepthwiseConv()
1130 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.first.GetInfo()); in AddConvLayerWithDepthwiseConv()
1621 auto biasTensor = CreateConstTensor(node.input(2)); in ParseBatchNormalization() local
1628 biasTensor.first, in ParseBatchNormalization()
1893 auto biasTensor = CreateConstTensor(node.input(2)); in ParseConv() local
1895 IConnectableLayer* biasLayer = m_Network->AddConstantLayer(biasTensor.first); in ParseConv()
1896 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.first.GetInfo()); in ParseConv()
/external/armnn/src/backends/backendsCommon/
DWorkloadData.cpp181 void ValidateBiasTensorQuantization(const TensorInfo& biasTensor, in ValidateBiasTensorQuantization() argument
199 if (biasTensor.GetQuantizationOffset() != 0) in ValidateBiasTensorQuantization()
202 to_string(biasTensor.GetQuantizationOffset())); in ValidateBiasTensorQuantization()
205 …if (biasTensor.HasMultipleQuantizationScales() || weightsTensorInfo.HasMultipleQuantizationScales(… in ValidateBiasTensorQuantization()
209 const std::vector<float>& biasScales = biasTensor.GetQuantizationScales(); in ValidateBiasTensorQuantization()
230 VerifyBiasQuantizationScale(biasTensor.GetQuantizationScale(), expectedScale); in ValidateBiasTensorQuantization()
/external/armnn/src/armnnDeserializer/
DDeserializer.cpp1513 armnn::ConstTensor biasTensor; in ParseConvolution2d() local
1533 biasTensor = ToConstTensor(flatBufferLayer->biases()); in ParseConvolution2d()
1534 auto biasLayer = m_Network->AddConstantLayer(biasTensor); in ParseConvolution2d()
1536 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.GetInfo()); in ParseConvolution2d()
2239 armnn::ConstTensor biasTensor = ToConstTensor(flatBufferLayer->biases()); in ParseFullyConnected() local
2240 auto biasLayer = m_Network->AddConstantLayer(biasTensor); in ParseFullyConnected()
2242 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensor.GetInfo()); in ParseFullyConnected()