serialize.hpp

// Copyright (C) 2018-2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
 
#pragma once
 
#include <string>
 
#include "ngraph/opsets/opset.hpp"
#include "ngraph/pass/pass.hpp"
#include "transformations_visibility.hpp"
 
namespace ngraph {
namespace pass {
 
class TRANSFORMATIONS_API Serialize;
 
}  // namespace pass
}  // namespace ngraph
 
/**
 * @ingroup ie_transformation_common_api
 * @brief Serialize transformation converts ngraph::Function into IR files
 * @attention
 * - dynamic shapes are not supported
 * - order of generated layers in xml file is ngraph specific (given by
 * get_ordered_ops()); MO generates file with different order, but they are
 * logically equivalent
 */
class ngraph::pass::Serialize : public ngraph::pass::FunctionPass {
public:
    enum class Version { IR_V10 };
    NGRAPH_RTTI_DECLARATION;
    bool run_on_function(std::shared_ptr<ngraph::Function> f) override;
 
    Serialize(std::ostream & xmlFile, std::ostream & binFile,
              Version version = Version::IR_V10,
              std::map<std::string, ngraph::OpSet> custom_opsets = {});
 
    Serialize(const std::string& xmlPath, const std::string& binPath,
              Version version = Version::IR_V10,
              std::map<std::string, ngraph::OpSet> custom_opsets = {});
 
private:
    std::ostream * m_xmlFile;
    std::ostream * m_binFile;
    const std::string m_xmlPath;
    const std::string m_binPath;
    const Version m_version;
    const std::map<std::string, ngraph::OpSet> m_custom_opsets;
};