Searched refs:detectionClassesInfo (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | DetectionPostProcessTestImpl.hpp | 161 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 D | RefDetectionPostProcessTests.cpp | 84 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 D | RefDetectionPostProcessWorkload.cpp | 42 const TensorInfo& detectionClassesInfo = GetTensorInfo(outputs[1]); in Execute() local 56 detectionBoxesInfo, detectionClassesInfo, in Execute()
|
H A D | DetectionPostProcess.cpp | 144 const TensorInfo& detectionClassesInfo, in DetectionPostProcess() argument 156 IgnoreUnused(anchorsInfo, detectionClassesInfo, detectionScoresInfo, numDetectionsInfo); in DetectionPostProcess()
|
H A D | DetectionPostProcess.hpp | 21 const TensorInfo& detectionClassesInfo,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | DetectionPostProcessEndToEndTestImpl.hpp | 26 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 D | OptimizerTests.cpp | 695 armnn::TensorInfo detectionClassesInfo({1, 3}, DataType::QAsymmU8); variable 711 layer->GetOutputSlot(2).SetTensorInfo(detectionClassesInfo);
|
H A D | ShapeInferenceTests.cpp | 351 const TensorShape detectionClassesInfo{ 1, 3, 4 }; variable
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
H A D | WorkloadData.cpp | 2754 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()
|