Home
last modified time | relevance | path

Searched refs:GetFirstLayerWithName (Results 1 – 10 of 10) sorted by relevance

/external/armnn/src/backends/neon/test/
DNeonFallbackTests.cpp68 armnn::Layer* const layer0 = GetFirstLayerWithName(graph, "input0");
69 armnn::Layer* const layer1 = GetFirstLayerWithName(graph, "input1");
70 armnn::Layer* const layer2 = GetFirstLayerWithName(graph, "input2");
71 armnn::Layer* const layer3 = GetFirstLayerWithName(graph, "add");
72 armnn::Layer* const layer4 = GetFirstLayerWithName(graph, "[ add (0) -> sub (1) ]");
73 armnn::Layer* const layer5 = GetFirstLayerWithName(graph, "sub");
74 armnn::Layer* const layer6 = GetFirstLayerWithName(graph, "output");
212 armnn::Layer* const layer0 = GetFirstLayerWithName(graph, "input0");
213 armnn::Layer* const layer1 = GetFirstLayerWithName(graph, "input1");
214 armnn::Layer* const layer2 = GetFirstLayerWithName(graph, "add");
[all …]
/external/armnn/src/backends/cl/test/
DClFallbackTests.cpp58 armnn::Layer* const layer0 = GetFirstLayerWithName(graph, "input0");
59 armnn::Layer* const layer1 = GetFirstLayerWithName(graph, "input1");
60 armnn::Layer* const layer2 = GetFirstLayerWithName(graph, "input2");
61 armnn::Layer* const layer3 = GetFirstLayerWithName(graph, "add");
62 armnn::Layer* const layer4 = GetFirstLayerWithName(graph, "[ add (0) -> sub (1) ]");
63 armnn::Layer* const layer5 = GetFirstLayerWithName(graph, "sub");
64 armnn::Layer* const layer6 = GetFirstLayerWithName(graph, "output");
204 armnn::Layer* const layer0 = GetFirstLayerWithName(graph, "input0");
205 armnn::Layer* const layer1 = GetFirstLayerWithName(graph, "input1");
206 armnn::Layer* const layer2 = GetFirstLayerWithName(graph, "input2");
[all …]
/external/armnn/src/armnn/test/
DGraphTests.cpp44 armnn::Layer* const layerA = GetFirstLayerWithName(graph, "layerA");
45 armnn::Layer* const layerB = GetFirstLayerWithName(graph, "layerB");
46 armnn::Layer* const layerC = GetFirstLayerWithName(graph, "layerC");
47 armnn::Layer* const layerO = GetFirstLayerWithName(graph, "output");
48 armnn::Layer* const layerE = GetFirstLayerWithName(graph, "layerE");
49 armnn::Layer* const layerD = GetFirstLayerWithName(graph, "layerD");
88 armnn::Layer* const layerA = GetFirstLayerWithName(graph, "layerA");
89 armnn::Layer* const layerB = GetFirstLayerWithName(graph, "layerB");
90 armnn::Layer* const layerC = GetFirstLayerWithName(graph, "layerC");
91 armnn::Layer* const layerD = GetFirstLayerWithName(graph, "layerD");
[all …]
/external/armnn/src/armnnTfLiteParser/test/
DDetectionPostProcess.cpp249 armnn::Layer* boxEncodingLayer = GetFirstLayerWithName(graph, "box_encodings");
254 armnn::Layer* scoresLayer = GetFirstLayerWithName(graph, "scores");
260 …armnn::Layer* detectionPostProcessLayer = GetFirstLayerWithName(graph, "DetectionPostProcess:0:0");
266 armnn::Layer* detectionBoxesLayer = GetFirstLayerWithName(graph, "detection_boxes");
271 armnn::Layer* detectionClassesLayer = GetFirstLayerWithName(graph, "detection_classes");
276 armnn::Layer* detectionScoresLayer = GetFirstLayerWithName(graph, "detection_scores");
281 armnn::Layer* numDetectionsLayer = GetFirstLayerWithName(graph, "num_detections");
/external/armnn/src/armnn/test/optimizations/
DAddBroadcastReshapeLayerTests.cpp57 Layer* const reshapeLayer = GetFirstLayerWithName(graph, reshapeLayerName); in AddBroadcastReshapeLayerOptimizerTest()
151 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:sub-0");
196 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:div-0");
241 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0");
285 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0");
334 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0");
394 Layer* const reshapeLayer1 = GetFirstLayerWithName(graph, "Reshape_for:add1-1");
395 Layer* const reshapeLayer2 = GetFirstLayerWithName(graph, "Reshape_for:add2-1");
DPermuteDepthwiseConv2dWeightsTests.cpp62 armnn::Layer* const permuteLayer = GetFirstLayerWithName(graph, "permute_layer"); in PermuteDepthwiseConv2dWeightsTestRunner()
DReduceMultipleAxesTests.cpp83 Layer* const reduceLayer = GetFirstLayerWithName(graph, layerName); in ReduceWithMultipleAxesTest()
DFuseActivationTests.cpp485 …Layer* fusedReceiverLayer = GetFirstLayerWithName(graphFused, "fused-activation-into-receiverLayer… in FuseActivationIntoPreviousLayerTest()
/external/armnn/src/armnnTestUtils/
DGraphUtils.hpp14 armnn::Layer* GetFirstLayerWithName(armnn::Graph& graph, const std::string& name);
DGraphUtils.cpp22 armnn::Layer* GetFirstLayerWithName(armnn::Graph& graph, const std::string& name) in GetFirstLayerWithName() function