Function BuildSme2ShapeProfile

src/armnn/Sme2ShapePolicy.cpp:296–337

}

Sme2ShapeProfile BuildSme2ShapeProfile(const Graph& graph, bool reduceFp32ToFp16)
{
    Sme2ShapeProfile profile;
    profile.m_HasFp16 = reduceFp32ToFp16;

    for (const Layer* layer : graph)
    {
        switch (layer->GetType())
        {
            case LayerType::Convolution2d:
                RecordConvolution2d(profile, *layer);
                break;
            case LayerType::FullyConnected:
                RecordFullyConnected(profile, *layer);
                break;
            case LayerType::BatchMatMul:
                RecordBatchMatMul(profile, *layer);
                break;
            case LayerType::DepthwiseConvolution2d:
                RecordDepthwiseConvolution2d(profile, *layer);
                break;
            default:
                for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
                {
                    if (layer->GetInputSlot(i).IsTensorInfoSet())
                    {
                        RecordTensorType(profile, layer->GetInputSlot(i).GetTensorInfo());
                    }
                }
                for (unsigned int i = 0; i < layer->GetNumOutputSlots(); ++i)
                {
                    if (layer->GetOutputSlot(i).IsTensorInfoSet())
                    {
                        RecordTensorType(profile, layer->GetOutputSlot(i).GetTensorInfo());
                    }
                }
                break;
        }
    }

    return profile;
}

unsigned int CapWorkerCount(unsigned int workers, unsigned int cap)
{