| /external/armnn/src/backends/neon/ |
| D | NeonBackend.cpp | 181 auto* activationLayer = PolymorphicDowncast<ActivationLayer*>(&child); in OptimizeSubgraphView() local
187 ActivationDescriptor activationDesc = activationLayer->GetParameters(); in OptimizeSubgraphView()
202 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
213 activationLayer, in OptimizeSubgraphView()
217 untouched.erase(activationLayer->GetGuid()); in OptimizeSubgraphView()
234 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
244 … activationLayer, in OptimizeSubgraphView()
248 untouched.erase(activationLayer->GetGuid()); in OptimizeSubgraphView()
265 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
275 activationLayer, in OptimizeSubgraphView()
[all …]
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeSubgraphUtils.hpp | 57 ActivationLayer* activationLayer, in FuseLayer() argument
63 SubgraphView substitutionSubgraph({baseLayer, activationLayer}, in FuseLayer()
65 CreateIOutputsFrom({activationLayer})); in FuseLayer()
76 ActivationLayer* activationLayer, in FuseAdditionLayer() argument
88 activationLayer, in FuseAdditionLayer()
97 ActivationLayer* activationLayer, in FuseSubtractionLayer() argument
109 activationLayer, in FuseSubtractionLayer()
118 ActivationLayer* activationLayer, in FuseDivisionLayer() argument
130 activationLayer, in FuseDivisionLayer()
139 ActivationLayer* activationLayer, in FuseMultiplicationLayer() argument
[all …]
|
| /external/armnn/src/backends/cl/ |
| D | ClBackend.cpp | 329 auto* activationLayer = PolymorphicDowncast<ActivationLayer*>(&child); in OptimizeSubgraphView() local
335 ActivationDescriptor activationDesc = activationLayer->GetParameters(); in OptimizeSubgraphView()
350 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
361 activationLayer, in OptimizeSubgraphView()
365 untouched.erase(activationLayer->GetGuid()); in OptimizeSubgraphView()
382 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
392 … activationLayer, in OptimizeSubgraphView()
396 untouched.erase(activationLayer->GetGuid()); in OptimizeSubgraphView()
413 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
423 activationLayer, in OptimizeSubgraphView()
[all …]
|
| /external/android-nn-driver/ |
| D | ConversionUtils.cpp | 149 armnn::IConnectableLayer* activationLayer = prevLayer; in ProcessActivation() local
208 activationLayer = data.m_Network->AddActivationLayer(activationDesc); in ProcessActivation()
209 activationLayer->SetBackendId(setBackend); in ProcessActivation()
211 prevLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in ProcessActivation()
212 activationLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in ProcessActivation()
215 return activationLayer; in ProcessActivation()
|
| /external/armnn/delegate/classic/src/ |
| D | Activation.hpp | 117 …armnn::IConnectableLayer* activationLayer = delegateData.m_Network->AddActivationLayer(activationD… in VisitActivationOperator() local
118 ARMNN_ASSERT(activationLayer != nullptr); in VisitActivationOperator()
120 armnn::IOutputSlot& outputSlot = activationLayer->GetOutputSlot(0); in VisitActivationOperator()
124 if(ProcessInputs(activationLayer,delegateData, tfLiteContext, tfLiteNode) != kTfLiteOk ) in VisitActivationOperator()
130 return Connect(activationLayer, tfLiteNode, delegateData); in VisitActivationOperator()
|
| D | ClassicDelegateUtils.hpp | 253 armnn::IConnectableLayer* activationLayer = data.m_Network->AddActivationLayer(activationDesc); in FusedActivation() local
254 activationLayer->SetBackendId(setBackend); in FusedActivation()
256 ARMNN_ASSERT(activationLayer != nullptr); in FusedActivation()
257 activationLayer->GetOutputSlot(0).SetTensorInfo(activationOutputInfo); in FusedActivation()
260 …for (unsigned int outputIndex = 0; outputIndex < activationLayer->GetNumOutputSlots(); ++outputInd… in FusedActivation()
263 tfLiteNode->outputs->data[outputIndex])]->Connect(activationLayer->GetInputSlot(0)); in FusedActivation()
264 armnn::IOutputSlot& outputSlot = activationLayer->GetOutputSlot(outputIndex); in FusedActivation()
|
| /external/armnn/src/armnn/test/ |
| D | DebugCallbackTest.cpp | 29 IConnectableLayer* activationLayer = net->AddActivationLayer(descriptor, "Activation:ReLu"); in CreateSimpleNetwork() local
33 input->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in CreateSimpleNetwork()
34 activationLayer->GetOutputSlot(0).Connect(output->GetInputSlot(0)); in CreateSimpleNetwork()
37 activationLayer->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 5 }, DataType::Float32)); in CreateSimpleNetwork()
|
| D | ShapeInferenceTests.cpp | 135 auto activationLayer = network->AddActivationLayer(descriptor, "activation"); variable
137 inputLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0));
138 …activationLayer->GetOutputSlot(0).SetTensorInfo({TensorShape{Dimensionality::NotSpecified}, DataTy…
140 CHECK_NOTHROW(activationLayer->GetOutputSlot(0).IsTensorInfoSet());
142 CHECK(activationLayer->GetOutputSlot(0).GetTensorInfo() == tensorInfo);
155 activationLayer = network->AddActivationLayer(descriptor, "activation");
157 inputLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0));
158 …activationLayer->GetOutputSlot(0).SetTensorInfo({TensorShape{Dimensionality::NotSpecified}, DataTy…
160 CHECK_NOTHROW(activationLayer->GetOutputSlot(0).IsTensorInfoSet());
167 activationLayer = network->AddActivationLayer(descriptor, "activation");
[all …]
|
| D | NetworkTests.cpp | 106 …armnn::IConnectableLayer* const activationLayer = net.AddActivationLayer(activationDesc, "activati… variable
107 CHECK(activationLayer);
109 poolingLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0));
115 activationLayer->GetOutputSlot(0).Connect(normalizationLayer->GetInputSlot(0));
235 CHECK(AreAllLayerInputSlotsConnected(*activationLayer));
247 checkOneOutputToOneInputConnection(poolingLayer, activationLayer);
248 checkOneOutputToOneInputConnection(activationLayer, normalizationLayer);
|
| /external/armnn/src/armnnSerializer/test/ |
| D | ActivationSerializationTests.cpp | 53 …armnn::IConnectableLayer* const activationLayer = network->AddActivationLayer(descriptor, "activat… variable
56 inputLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0));
59 activationLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0));
60 activationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | EndToEndTestImpl.hpp | 850 IConnectableLayer* activationLayer = net->AddActivationLayer(descriptor); in ForceImportWithAlignedBuffersEndToEndTest() local
852 input->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in ForceImportWithAlignedBuffersEndToEndTest()
853 activationLayer->GetOutputSlot(0).Connect(output->GetInputSlot(0)); in ForceImportWithAlignedBuffersEndToEndTest()
855 activationLayer->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32)); in ForceImportWithAlignedBuffersEndToEndTest()
952 IConnectableLayer* activationLayer = net->AddActivationLayer(descriptor); in ForceImportWithMisalignedInputBuffersEndToEndTest() local
956 input->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in ForceImportWithMisalignedInputBuffersEndToEndTest()
957 activationLayer->GetOutputSlot(0).Connect(output->GetInputSlot(0)); in ForceImportWithMisalignedInputBuffersEndToEndTest()
959 activationLayer->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32)); in ForceImportWithMisalignedInputBuffersEndToEndTest()
1075 IConnectableLayer* activationLayer = net->AddActivationLayer(descriptor); in ForceImportWithMisalignedOutputBuffersEndToEndTest() local
1079 input->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in ForceImportWithMisalignedOutputBuffersEndToEndTest()
[all …]
|
| /external/armnn/delegate/opaque/src/ |
| D | OpaqueDelegateUtils.hpp | 294 armnn::IConnectableLayer* activationLayer = data.m_Network->AddActivationLayer(activationDesc); in FusedActivation() local
295 activationLayer->SetBackendId(setBackend); in FusedActivation()
297 ARMNN_ASSERT(activationLayer != nullptr); in FusedActivation()
298 activationLayer->GetOutputSlot(0).SetTensorInfo(activationOutputInfo); in FusedActivation()
312 …for (unsigned int outputIndex = 0; outputIndex < activationLayer->GetNumOutputSlots(); ++outputInd… in FusedActivation()
315 outputIndexArray[outputIndex])]->Connect(activationLayer->GetInputSlot(0)); in FusedActivation()
317 armnn::IOutputSlot& outputSlot = activationLayer->GetOutputSlot(outputIndex); in FusedActivation()
|
| /external/armnn/shim/sl/canonical/ |
| D | ConversionUtils.cpp | 894 armnn::IConnectableLayer* activationLayer = prevLayer; in ProcessActivation() local
953 activationLayer = data.m_Network->AddActivationLayer(activationDesc); in ProcessActivation()
954 activationLayer->SetBackendId(setBackend); in ProcessActivation()
956 prevLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in ProcessActivation()
957 activationLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in ProcessActivation()
960 return activationLayer; in ProcessActivation()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | FuseActivationTests.cpp | 403 IConnectableLayer* activationLayer = network->AddActivationLayer(activationDescriptor, in CreateNetwork() local
430 activationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); in CreateNetwork()
434 receiverLayer->GetOutputSlot(0).Connect(activationLayer->GetInputSlot(0)); in CreateNetwork()
435 activationLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); in CreateNetwork()
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 4713 IConnectableLayer* activationLayer = in AddFusedActivationLayer() local
4717 prevOutputSlot.Connect(activationLayer->GetInputSlot(0)); in AddFusedActivationLayer()
4718 activationLayer->GetOutputSlot(0).SetTensorInfo(prevOutputSlot.GetTensorInfo()); in AddFusedActivationLayer()
4719 return activationLayer; in AddFusedActivationLayer()
|