Function SelectNumberOfThreads

src/armnn/Sme2ShapePolicy.cpp:412–435

}

unsigned int SelectNumberOfThreads(const Sme2ShapeProfile& profile, unsigned int requestedThreads)
{
    if (!profile.m_HasQuantized || ShouldDisableSme(profile))
    {
        return requestedThreads;
    }

    if (profile.m_GemmLikeOps == 0)
    {
        return CapWorkerCount(requestedThreads, 1);
    }

    if (profile.m_HasSegmentationShape || profile.m_HasStyleTransferShape)
    {
        return requestedThreads;
    }

    if (profile.m_HasPoseShape)
    {
        return CapWorkerCount(requestedThreads, 4);
    }

    return CapWorkerCount(requestedThreads, 1);
}

} // namespace