ClContextSerializer.hpp

Go to the documentation of this file.

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

#pragma once

#include <arm_compute/core/CL/CLCompileContext.h>

#include <flatbuffers/flatbuffers.h>

namespace armnn
{

class ClContextSerializer
{
public:
    ClContextSerializer()  = default;
    ~ClContextSerializer() = default;

    /// Serializes the CLCompileContext built-in programs
    /// @param [in] clCompileContext The CLCompileContext to be serialized.
    void Serialize(const arm_compute::CLCompileContext& clCompileContext);

    /// Serializes the ClContext to the stream.
    /// @param [stream] the stream to save to
    /// @return true if ClContext is Serialized to the Stream, false otherwise
    bool SaveSerializedToStream(std::ostream& stream);

private:
    /// FlatBufferBuilder to create the CLContext FlatBuffers.
    flatbuffers::FlatBufferBuilder m_FlatBufferBuilder;
};

} // namespace armnn