Method OptimizeSubgraphView

src/backends/tosaReference/TosaRefBackend.cpp:82–186

}

OptimizationViews TosaRefBackend::OptimizeSubgraphView(const SubgraphView& subgraph,
                                                       const ModelOptions& modelOptions) const
{
    OptimizationViews optimizationViews(modelOptions);

    auto handler = std::make_unique<TosaSerializationHandler>();

    std::vector<std::string> graphInputs;
    std::vector<std::string> graphOutputs;
    std::vector<TosaSerializationOperator*> operators;
    std::vector<TosaSerializationTensor*> tensors;
    OpBlockSequencer<Layer, TosaSerializationBasicBlock> sequencer;

    // These sets are created to check the duplication of tensor and input
    std::unordered_set<std::string> graphInputsSet;
    std::unordered_set<std::string> uniqueTensorNamesSet;

    auto it = subgraph.begin();
    while (it != subgraph.end())
    {
        Layer& base = *(PolymorphicDowncast<Layer*>(*it));
        ++it;

        if (base.GetType() == LayerType::Input ||
            base.GetType() == LayerType::Output)
        {
            continue;
        }

        TosaSerializationBasicBlock* mappings = GetTosaMappingFromLayer(&base);

        // Loop through inputs to see if there are any graph inputs, if so save them.
        // If it's an input to the graph "input" can be found in the string.
        for (const std::string& blockInputName : mappings->GetInputs())
        {
            if ((blockInputName.find("input") != std::string::npos) && !graphInputsSet.count(blockInputName))
            {
                graphInputs.push_back(blockInputName);
                graphInputsSet.insert(blockInputName);
            }
        }

        // Loop through outputs to see if there are any graph outputs, if so save them.
        // If it's an output to the graph "output" can be found in the string.
        for (const std::string& blockOutputName : mappings->GetOutputs())
        {
            if (blockOutputName.find("output") != std::string::npos)
            {
                graphOutputs.push_back(blockOutputName);
            }
        }

        sequencer.Add(&base, mappings);
    }

    for (auto & pair : sequencer.Finish())
    {
        auto blockOperators = pair.block->GetOperators();