Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/
H A DDetectionPostProcess.cpp260 …armnn::Layer* detectionPostProcessLayer = GetFirstLayerWithName(graph, "DetectionPostProcess:0:0"); variable
261 CHECK((detectionPostProcessLayer->GetType() == armnn::LayerType::DetectionPostProcess));
262 CHECK(CheckNumberOfInputSlot(detectionPostProcessLayer, 2));
263 CHECK(CheckNumberOfOutputSlot(detectionPostProcessLayer, 4));
296 CHECK(IsConnected(boxEncodingLayer, detectionPostProcessLayer, 0, 0, boxEncodingTensor));
297 CHECK(IsConnected(scoresLayer, detectionPostProcessLayer, 0, 1, scoresTensor));
298 CHECK(IsConnected(detectionPostProcessLayer, detectionBoxesLayer, 0, 0, detectionBoxesTensor));
299 …CHECK(IsConnected(detectionPostProcessLayer, detectionClassesLayer, 1, 0, detectionClassesTensor));
300 … CHECK(IsConnected(detectionPostProcessLayer, detectionScoresLayer, 2, 0, detectionScoresTensor));
301 CHECK(IsConnected(detectionPostProcessLayer, numDetectionsLayer, 3, 0, numDetectionsTensor));