Searched refs:detectionClassesLayer (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/ |
| H A D | DetectionPostProcess.cpp | 271 armnn::Layer* detectionClassesLayer = GetFirstLayerWithName(graph, "detection_classes"); variable
272 CHECK((detectionClassesLayer->GetType() == armnn::LayerType::Output));
273 CHECK(CheckNumberOfInputSlot(detectionClassesLayer, 1));
274 CHECK(CheckNumberOfOutputSlot(detectionClassesLayer, 0));
299 …CHECK(IsConnected(detectionPostProcessLayer, detectionClassesLayer, 1, 0, detectionClassesTensor));
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | DetectionPostProcessEndToEndTestImpl.hpp | 50 armnn::IConnectableLayer* detectionClassesLayer = net->AddOutputLayer(1, "detectionClasses"); in CreateDetectionPostProcessNetwork() local
56 Connect(detectionLayer, detectionClassesLayer, detectionClassesInfo, 1, 0); in CreateDetectionPostProcessNetwork()
|