xref: /aosp_15_r20/external/armnn/src/armnn/OptimizedNetworkImpl.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)

//
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include <common/include/ProfilingGuid.hpp>

namespace armnn
{

class OptimizedNetworkImpl
{
public:
    OptimizedNetworkImpl(const OptimizedNetworkImpl& other, const ModelOptions& modelOptions);
    OptimizedNetworkImpl(std::unique_ptr<Graph> graph);
    OptimizedNetworkImpl(std::unique_ptr<Graph> graph, const ModelOptions& modelOptions);
    virtual ~OptimizedNetworkImpl();

    virtual Status PrintGraph();
    virtual Status SerializeToDot(std::ostream& stream) const;

    virtual arm::pipe::ProfilingGuid GetGuid() const { return m_Guid; };

    virtual size_t GetNumInputs() const;
    virtual size_t GetNumOutputs() const;

    Graph& GetGraph() { return *m_Graph; }
    Graph& GetGraph() const { return *m_Graph; }
    ModelOptions& GetModelOptions() { return m_ModelOptions; }

    void ExecuteStrategy(IStrategy& strategy) const;

private:
    std::unique_ptr<Graph> m_Graph;
    arm::pipe::ProfilingGuid m_Guid;
    ModelOptions m_ModelOptions;
};

}