ISerializer.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once

#include "armnn/INetwork.hpp"
#include "armnn/NetworkFwd.hpp"
#include "armnn/Types.hpp"

namespace armnnSerializer
{

class ISerializer;
using ISerializerPtr = std::unique_ptr<ISerializer, void(*)(ISerializer* serializer)>;

class ISerializer
{
public:
    static ISerializer* CreateRaw();
    static ISerializerPtr Create();
    static void Destroy(ISerializer* serializer);

    virtual void Serialize(const armnn::INetwork& inNetwork) = 0;

    virtual bool SaveSerializedToStream(std::ostream& stream) = 0;

protected:
    virtual ~ISerializer() {}
};

} //namespace armnnSerializer