Home
last modified time | relevance | path

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

/external/executorch/backends/apple/mps/runtime/operations/
DConvolutionOps.mm59 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input3_id());
60 biasTensor = [_mpsGraph expandDimsOfTensor:biasTensor
64 secondaryTensor:biasTensor
120 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input3_id());
121 biasTensor = [_mpsGraph expandDimsOfTensor:biasTensor
125 secondaryTensor:biasTensor
DLinearAlgebra.mm46 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input1_id());
66 MPSGraphTensor* betaBiasTensor = biasTensor;
71 betaBiasTensor = [_mpsGraph multiplicationWithPrimaryTensor:biasTensor
DNormalizationOps.mm34 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->bias_id());
55 MPSGraphTensor* reshapedBiasTensor = [_mpsGraph reshapeTensor:biasTensor
93 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->bias_id());
115 betaTensor:biasTensor
/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/pytorch/aten/src/ATen/native/mps/operations/
DLinear.mm255 MPSGraphTensor* biasTensor = nil;
258 biasTensor = [mpsGraph reductionSumWithTensor:gradOutputTensor axis:0 name:nil];
265 newCachedGraph->biasTensor_ = biasTensor;
DConvolution.mm306 MPSGraphTensor* biasTensor = nil;
308 biasTensor = mpsGraphUnrankedPlaceHolder(mpsGraph, getMPSDataType(bias_opt.value()));
316 …outputTensor = [mpsGraph additionWithPrimaryTensor:outputTensor secondaryTensor:biasTensor name:ni…
320 newCachedGraph->biasTensor_ = biasTensor;
DLinearAlgebra.mm465 MPSGraphTensor* biasTensor = mpsGraphRankedPlaceHolder(mpsGraph, *bias_);
478 auto biasTimesBetaTensor = biasTensor;
482 biasTimesBetaTensor = [mpsGraph multiplicationWithPrimaryTensor:biasTensor
496 newCachedGraph->biasTensor_ = biasTensor;
DNormalization.mm162 MPSGraphTensor* biasTensor = nil;
164biasTensor = mpsGraphRankedPlaceHolder(mpsGraph, getMPSDataType(bias_opt.value()), new_mean_shape);
269 betaTensor:biasTensor
289 newCachedGraph->biasTensor_ = biasTensor;
DRnnOps.mm26 * stateTensor, cellStateTensor, recurrentWeight, inputWeight, biasTensor
/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()