class ngraph::pass::Serialize

Overview

Serialize transformation converts ngraph::Function into IR files. More…

#include <serialize.hpp>

class Serialize: public ngraph::pass::FunctionPass
{
public:
    // enums

    enum Version;

    // construction

    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 = {}
        );

    // methods

    virtual bool run_on_function(std::shared_ptr<ngraph::Function> f);
};

Inherited Members

public:
    // typedefs

    typedef DiscreteTypeInfo type_info_t;

    // methods

    bool get_property(const PassPropertyMask& prop_mask) const;
    void set_name(const std::string& name);
    std::string get_name() const;
    void set_callback(const param_callback& callback);
    virtual void set_pass_config(const std::shared_ptr<PassConfig>& pass_config);
    std::shared_ptr<PassConfig> get_pass_config();
    bool m_transformation_callback(const std::shared_ptr<const Node>& node);
    bool transformation_callback(const std::shared_ptr<const Node>& node);
    virtual const type_info_t& get_type_info() const = 0;
    virtual bool run_on_function(std::shared_ptr<ngraph::Function>) = 0;

Detailed Documentation

Serialize transformation converts ngraph::Function into IR files.

  • 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