Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DDetectionPostProcessTestImpl.hpp161 armnn::TensorInfo detectionClassesInfo({ 1, 3 }, armnn::DataType::Float32); in DetectionPostProcessImpl() local
166 std::vector<float> actualDetectionClassesOutput(detectionClassesInfo.GetNumElements()); in DetectionPostProcessImpl()
174 auto classesHandle = tensorHandleFactory.CreateTensorHandle(detectionClassesInfo); in DetectionPostProcessImpl()
199 AddOutputToWorkload(data, info, detectionClassesInfo, classesHandle.get()); in DetectionPostProcessImpl()
233 detectionClassesInfo.GetShape()); in DetectionPostProcessImpl()
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A DRefDetectionPostProcessTests.cpp84 armnn::TensorInfo detectionClassesInfo({ 1, 3 }, armnn::DataType::Float32); in DetectionPostProcessTestImpl() local
133 std::vector<float> detectionClasses(detectionClassesInfo.GetNumElements()); in DetectionPostProcessTestImpl()
140 detectionClassesInfo, in DetectionPostProcessTestImpl()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefDetectionPostProcessWorkload.cpp42 const TensorInfo& detectionClassesInfo = GetTensorInfo(outputs[1]); in Execute() local
56 detectionBoxesInfo, detectionClassesInfo, in Execute()
H A DDetectionPostProcess.cpp144 const TensorInfo& detectionClassesInfo, in DetectionPostProcess() argument
156 IgnoreUnused(anchorsInfo, detectionClassesInfo, detectionScoresInfo, numDetectionsInfo); in DetectionPostProcess()
H A DDetectionPostProcess.hpp21 const TensorInfo& detectionClassesInfo,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDetectionPostProcessEndToEndTestImpl.hpp26 armnn::TensorInfo detectionClassesInfo({ 1, 3 }, armnn::DataType::Float32); in CreateDetectionPostProcessNetwork() local
56 Connect(detectionLayer, detectionClassesLayer, detectionClassesInfo, 1, 0); in CreateDetectionPostProcessNetwork()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DOptimizerTests.cpp695 armnn::TensorInfo detectionClassesInfo({1, 3}, DataType::QAsymmU8); variable
711 layer->GetOutputSlot(2).SetTensorInfo(detectionClassesInfo);
H A DShapeInferenceTests.cpp351 const TensorShape detectionClassesInfo{ 1, 3, 4 }; variable
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp2754 const TensorInfo& detectionClassesInfo = workloadInfo.m_OutputTensorInfos[1]; in Validate() local
2778 ValidateTensorNumDimensions(detectionClassesInfo, descriptorName, 2, "detection classes"); in Validate()
2784 …ValidateTensorDataType(detectionClassesInfo, DataType::Float32, descriptorName, "detection classes… in Validate()