Home
last modified time | relevance | path

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

/external/armnn/src/backends/neon/workloads/
DNeonDetectionPostProcessWorkload.cpp36 const TensorInfo& detectionClasses, in NeonDetectionPostProcessValidate() argument
56 armcomputetensorutils::BuildArmComputeTensorInfo(detectionClasses); in NeonDetectionPostProcessValidate()
94 arm_compute::ITensor& detectionClasses = AclTensorRef(m_Data.m_Outputs[1]); in NeonDetectionPostProcessWorkload() local
99 &detectionBoxes, &detectionClasses, &detectionScores, &numDetections, in NeonDetectionPostProcessWorkload()
DNeonDetectionPostProcessWorkload.hpp20 const TensorInfo& detectionClasses,
/external/armnn/src/backends/reference/test/
DRefDetectionPostProcessTests.cpp133 std::vector<float> detectionClasses(detectionClassesInfo.GetNumElements()); in DetectionPostProcessTestImpl() local
148 detectionClasses.data(), in DetectionPostProcessTestImpl()
162 BOOST_CHECK_EQUAL_COLLECTIONS(detectionClasses.begin(), in DetectionPostProcessTestImpl()
163 detectionClasses.end(), in DetectionPostProcessTestImpl()
/external/armnn/src/backends/reference/workloads/
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()
DRefDetectionPostProcessWorkload.cpp39 float* detectionClasses = GetOutputTensorData<float>(1, m_Data); in Execute() local
47 detectionClasses, detectionScores, numDetections); in Execute()
DDetectionPostProcess.hpp29 float* detectionClasses,
/external/armnn/tests/
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()
/external/armnn/src/armnnTfLiteParser/test/
DDetectionPostProcess.cpp198 UnquantizedContainer detectionClasses = { 1.0f, 0.0f, 0.0f }; in BOOST_FIXTURE_TEST_CASE() local
218 { "detection_classes", detectionClasses}, in BOOST_FIXTURE_TEST_CASE()
/external/armnn/src/backends/reference/
DRefLayerSupport.hpp110 const TensorInfo& detectionClasses,
DRefLayerSupport.cpp715 const TensorInfo& detectionClasses, in IsDetectionPostProcessSupported() argument
721 …IgnoreUnused(anchors, detectionBoxes, detectionClasses, detectionScores, numDetections, descriptor… in IsDetectionPostProcessSupported()
/external/armnn/include/armnn/
DILayerSupport.hpp126 const TensorInfo& detectionClasses,
/external/armnn/src/backends/backendsCommon/
DLayerSupportBase.hpp112 const TensorInfo& detectionClasses,
DWorkloadFactory.cpp285 const TensorInfo& detectionClasses = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported() local
294 detectionClasses, in IsLayerConfigurationSupported()