Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DDetectionPostProcess.cpp111 float* detectionClasses, in AllocateOutputData() argument
121 detectionClasses[i] = armnn::numeric_cast<float>(selectedClasses[outputIndices[i]]); in AllocateOutputData()
130 detectionClasses[i] = 0.0f; in AllocateOutputData()
152 float* detectionClasses, in DetectionPostProcess() argument
278 detectionBoxes, detectionScores, detectionClasses, numDetections); in DetectionPostProcess()
318 detectionBoxes, detectionScores, detectionClasses, numDetections); in DetectionPostProcess()
H A DRefDetectionPostProcessWorkload.cpp51 float* detectionClasses = reinterpret_cast<float*>(outputs[1]->Map()); in Execute() local
59 detectionClasses, detectionScores, numDetections); in Execute()
H A DDetectionPostProcess.hpp29 float* detectionClasses,
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonDetectionPostProcessWorkload.cpp36 const TensorInfo& detectionClasses, in NeonDetectionPostProcessValidate() argument
56 armcomputetensorutils::BuildArmComputeTensorInfo(detectionClasses); in NeonDetectionPostProcessValidate()
100 arm_compute::ITensor& detectionClasses = AclTensorRef(m_Data.m_Outputs[1]); in NeonDetectionPostProcessWorkload() local
105 &detectionBoxes, &detectionClasses, &detectionScores, &numDetections, in NeonDetectionPostProcessWorkload()
H A DNeonDetectionPostProcessWorkload.hpp20 const TensorInfo& detectionClasses,
/aosp_15_r20/external/armnn/src/backends/reference/test/
H A DRefDetectionPostProcessTests.cpp133 std::vector<float> detectionClasses(detectionClassesInfo.GetNumElements()); in DetectionPostProcessTestImpl() local
148 detectionClasses.data(), in DetectionPostProcessTestImpl()
160 CHECK(std::equal(detectionClasses.begin(), detectionClasses.end(), in DetectionPostProcessTestImpl()
/aosp_15_r20/external/armnn/tests/
H A DMobileNetSsdDatabase.hpp108 std::vector<float> detectionClasses; in GetTestCaseData() local
118 detectionClasses.push_back(expectedObject.m_Class); in GetTestCaseData()
127 expectedOutputs.push_back(detectionClasses); in GetTestCaseData()
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/
H A DDetectionPostProcess.cpp195 UnquantizedContainer detectionClasses = { 1.0f, 0.0f, 0.0f }; variable
215 { "detection_classes", detectionClasses},
/aosp_15_r20/external/armnn/src/armnn/
H A DILayerSupport.cpp713 const TensorInfo& detectionClasses, in IsDetectionPostProcessSupported() argument
723 detectionClasses, in IsDetectionPostProcessSupported()
H A DBackendHelper.cpp518 const TensorInfo& detectionClasses, in IsDetectionPostProcessSupported() argument
524 …TensorInfos infos{boxEncodings, scores, anchors, detectionBoxes, detectionClasses, detectionScores… in IsDetectionPostProcessSupported()
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefLayerSupport.hpp127 const TensorInfo& detectionClasses,
H A DRefLayerSupport.cpp1301 const TensorInfo& detectionClasses, in IsDetectionPostProcessSupported() argument
1307 …IgnoreUnused(anchors, detectionBoxes, detectionClasses, detectionScores, numDetections, descriptor… in IsDetectionPostProcessSupported()
/aosp_15_r20/external/armnn/include/armnn/
H A DBackendHelper.hpp146 const TensorInfo& detectionClasses,
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DILayerSupport.hpp175 const TensorInfo& detectionClasses,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DLayerSupportBase.hpp141 const TensorInfo& detectionClasses,
H A DWorkloadFactory.cpp381 const TensorInfo& detectionClasses = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported() local
390 detectionClasses, in IsLayerConfigurationSupported()