Searched refs:m_NumClasses (Results 1 – 17 of 17) sorted by relevance
219 unsigned int numClassesWithBg = desc.m_NumClasses + 1; in DetectionPostProcess()246 for (unsigned int c = 0; c < desc.m_NumClasses; ++c) in DetectionPostProcess()285 unsigned int numClassesPerBox = std::min(desc.m_MaxClassesPerDetection, desc.m_NumClasses); in DetectionPostProcess()295 std::vector<unsigned int> maxScoreIndices = GenerateRangeK(desc.m_NumClasses); in DetectionPostProcess()297 decodedScores.data() + scoreIndex, desc.m_NumClasses); in DetectionPostProcess()
514 , m_NumClasses(0) in DetectionPostProcessDescriptor()529 m_NumClasses == rhs.m_NumClasses && in operator ==()548 uint32_t m_NumClasses; member
23 desc.m_NumClasses, in MakeInfo()
94 desc.m_NumClasses = 2; in DetectionPostProcessTestImpl()
34 desc.m_NumClasses = 2; in CreateDetectionPostProcessNetwork()
146 descriptor.m_NumClasses = 2u; in GenerateDescriptor()
175 detectPostProcess.Int("num_classes", descriptor.m_NumClasses); in GenerateDetectionPostProcessJsonString()
361 uint32_t m_NumClasses; member
194 data.m_Parameters.m_NumClasses = 2; in DetectionPostProcessImpl()
132 assert desc.m_NumClasses == 0
213 fn("NumClasses", std::to_string(desc.m_NumClasses)); in Serialize()
366 descriptor.m_NumClasses = 2; in BOOST_AUTO_TEST_CASE()
418 descriptor.m_NumClasses, in VisitDetectionPostProcessLayer()
2686 if (scoresInfo.GetShape()[2] != m_Parameters.m_NumClasses + 1) in Validate()
2392 desc.m_NumClasses = m["num_classes"].AsUInt32(); in ParseDetectionPostProcess()
1375 descriptor.m_NumClasses = flatBufferDescriptor->numClasses(); in ParseDetectionPostProcess()
1015 descriptor.m_NumClasses = 2; in BOOST_AUTO_TEST_CASE()