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()
|